本来想交poj 3070,结果大晚上的poj维护。。。想到自己oj上也有一道这样的题,线代学了矩阵,今天趁热学了学矩阵快速幂,推荐一个博客:
矩阵快速幂入门
感觉这道题真的不难,题干中给出了递推过程,看到网上都是求[1,1,1,0]的n次方,但是其实求到n-1次方就行了啊,以下是推导过程:
在自己oj上 求到n次方可以过,只求到n-1次方反而超时,结果其实是一样的,很难受了这波,是数据太水还是另有原因,明天从poj上测一测就知道了。。。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
using namespace std;
#define mod 1000000007
struct node
{
long long m[2][2];
}ans;
node mulp(node a, node b)
{
node temp;
memset(temp.m, 0, sizeof(temp.m));
for(int i = 0; i < 2; i ++)
for(int j = 0; j < 2; j ++){
for(int k = 0; k < 2; k ++)
temp.m[i][j] = (temp.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
}
return temp;
}
long long matrix_power(node a, long long b)
{
ans.m[0][0] = ans.m[1][1] = 1;
ans.m[0][1] = ans.m[1][0] = 0;
while(b)
{
if(b & 1) ans = mulp(ans, a);
a = mulp(a, a);
b >>= 1;
}
return ans.m[0][1];
}
int main()
{
long long n;
while(scanf("%lld",&n) != EOF)
{
node base = {1, 1, 1, 0};
printf("%lld\n",matrix_power(base, n));
}
return 0;
}