欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

牛客小白月赛20A

程序员文章站 2022-06-03 19:29:03
...

斐波那契

牛客小白月赛20A
题解:
牛客小白月赛20A
快速幂就过了

#include<stdio.h>
#include<string.h>
using namespace std;
const int N=2;
typedef long long ll;
struct Matrix {
	ll matrix[N][N];
};
const int mod = 1e9 + 7;
void init(Matrix &res) {
	memset(res.matrix,0,sizeof(res.matrix));
	for(int i=0; i<N; i++)
		res.matrix[i][i]=1;
}
Matrix multiplicative(Matrix a,Matrix b) {
	Matrix res;
	memset(res.matrix,0,sizeof(res.matrix));
	for(int i = 0 ; i < N ; i++) {
		for(int j = 0 ; j < N ; j++) {
			for(int k = 0 ; k < N ; k++) {
				res.matrix[i][j] += a.matrix[i][k]*b.matrix[k][j]%mod;
				res.matrix[i][j] = (res.matrix[i][j]+mod)%mod;
			}
		}
	}
	return res;
}
Matrix Pow(Matrix mx,ll m) {
	Matrix res,base=mx;
	init(res);
	while(m) {
		if(m&1)
			res=multiplicative(res,base);
		base=multiplicative(base,base);
		m>>=1;
	}
	return res;
}
int main(){
	ll n;
	scanf("%lld",&n);
	Matrix m,ans;
	m.matrix[0][0]=1,m.matrix[0][1]=1;
	m.matrix[1][0]=1,m.matrix[1][1]=0;
	ans=Pow(m,n-1);
	//printf("%lld %lld\n%lld %lld\n",ans.matrix[0][0],ans.matrix[0][1],ans.matrix[1][0],ans.matrix[1][1]);
	ll ss=(ans.matrix[0][0]%mod*((ans.matrix[0][0]%mod+ans.matrix[1][0]%mod))%mod)%mod;
	printf("%lld\n",ss);
	return 0;
}
相关标签: 几何