牛客小白月赛20A
程序员文章站
2022-06-03 19:29:03
...
斐波那契
题解:
快速幂就过了
#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;
}