(矩阵快速幂模板)爬楼梯再加强版

题目链接:http://sdnuoj.rainng.com/problem/show/1085

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 3;
const int mod = 1000000007;
struct mat{
   
	ll m[maxn][maxn];
};

mat mul(mat x, mat y, ll len) {
   
	mat tmp;
	for(int i=0; i<len; i++) {
   
		for(int j=0; j<len; j++) {
   
			tmp.m[i][j]=0;
			for(int k=0; k <len; k++) {
   
				tmp.m[i][j]+=(x.m[i][k]*y.m[k][j])%mod;
			}
			tmp.m[i][j]%=mod;
		}
	}
	return tmp;
}

mat matpow(mat x, mat y, ll num) {
   
	while(num>0) {
   
		if(num&1) {
   
			y=mul(y,x,3); 
	}
		x=mul(x,x,3);
		num=num>>1;
	}
	return y;
}

int main()
{
   
	mat res;
	ll n;
	while(~scanf("%lld", &n))
	{
   
		for(int i=0; i<maxn; i++)
			for(int j=0; j<maxn; j++)
				res.m[i][j]=0;
		for(int i=0; i<maxn; i++)
			res.m[0][i]=1;
		res.m[1][0]=1;
		res.m[2][1]=1;
		if(n==1)
            printf("1\n");
        else if(n==2)
            printf("2\n");
        else if(n==3)
            printf("4\n");
		res=matpow(res,res,n-2);
		printf("%lld\n",((4*res.m[2][0])%mod+(2*res.m[2][1])%mod+res.m[2][2]%mod)%mod);
		
	}
	return 0;
}
全部评论

相关推荐

评论
1
收藏
分享

创作者周榜

更多
正在热议
更多
# 春招至今,你的战绩如何? #
9672次浏览 89人参与
# 你的实习产出是真实的还是包装的? #
1742次浏览 40人参与
# MiniMax求职进展汇总 #
23851次浏览 308人参与
# 军工所铁饭碗 vs 互联网高薪资,你会选谁 #
7464次浏览 43人参与
# 简历第一个项目做什么 #
31568次浏览 330人参与
# 重来一次,我还会选择这个专业吗 #
433365次浏览 3926人参与
# 巨人网络春招 #
11304次浏览 223人参与
# 当下环境,你会继续卷互联网,还是看其他行业机会 #
186990次浏览 1122人参与
# 牛客AI文生图 #
21408次浏览 238人参与
# 不考虑薪资和职业,你最想做什么工作呢? #
152293次浏览 887人参与
# 研究所笔面经互助 #
118873次浏览 577人参与
# 简历中的项目经历要怎么写? #
310079次浏览 4194人参与
# AI时代,哪些岗位最容易被淘汰 #
63432次浏览 804人参与
# 面试紧张时你会有什么表现? #
30490次浏览 188人参与
# 你今年的平均薪资是多少? #
213013次浏览 1039人参与
# 你怎么看待AI面试 #
179875次浏览 1235人参与
# 高学历就一定能找到好工作吗? #
64313次浏览 620人参与
# 你最满意的offer薪资是哪家公司? #
76438次浏览 374人参与
# 我的求职精神状态 #
447988次浏览 3128人参与
# 正在春招的你,也参与了去年秋招吗? #
363243次浏览 2637人参与
# 腾讯音乐求职进展汇总 #
160585次浏览 1111人参与
# 校招笔试 #
470477次浏览 2964人参与
牛客网
牛客网在线编程
牛客网题解
牛客企业服务