Another kind of Fibonacci
Time Limit: 3000/1000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)Total Submission(s): 2370 Accepted Submission(s): 945
Problem Description
As we all known , the Fibonacci series : F(0) = 1, F(1) = 1, F(N) = F(N - 1) + F(N - 2) (N >= 2).Now we define another kind of Fibonacci : A(0) = 1 , A(1) = 1 , A(N) = X * A(N - 1) + Y * A(N - 2) (N >= 2).And we want to Calculate S(N) , S(N) = A(0)
2
+A(1)
2+……+A(n)
2.
Input
There are several test cases.
Each test case will contain three integers , N, X , Y .
N : 2<= N <= 2 31 – 1
X : 2<= X <= 2 31– 1
Y : 2<= Y <= 2 31 – 1
Each test case will contain three integers , N, X , Y .
N : 2<= N <= 2 31 – 1
X : 2<= X <= 2 31– 1
Y : 2<= Y <= 2 31 – 1
Output
For each test case , output the answer of S(n).If the answer is too big , divide it by 10007 and give me the reminder.
Sample Input
2 1 1 3 2 3
Sample Output
6 196
思路:主要能推出递推式就好办了……还是比较好写的
但是这个题卡常数吧,,用long long 会TLE…
#include
#include
#include
#include
#include
using namespace std; typedef long long LL; const int N = 6; int k; int MOD; struct Mat { int mat[N][N]; Mat operator *(const Mat &a)const { Mat tmp; for(int i = 0; i < k; i++) for(int j = 0; j < k; j++) { tmp.mat[i][j] = 0; for(int l = 0; l < k; l++) tmp.mat[i][j] += mat[i][l] * a.mat[l][j] % MOD; tmp.mat[i][j] %= MOD; } return tmp; } }; int quick(int n,int x,int y) { if(n == 1)return 2; if(n == 0)return 1; n--; x %= MOD; y %= MOD; Mat ans; memset(ans.mat,0,sizeof(ans.mat)); for(int i = 0; i < k; i++)ans.mat[i][i] = 1; Mat tmp; memset(tmp.mat,0,sizeof(tmp.mat)); tmp.mat[0][0] = tmp.mat[5][0] = x*x%MOD; tmp.mat[0][1] = tmp.mat[5][1] = y*y%MOD; tmp.mat[0][2] = tmp.mat[5][2] = 2*x*y%MOD; tmp.mat[2][0] = tmp.mat[3][3] = x; tmp.mat[2][2] = tmp.mat[3][4] = y; tmp.mat[1][0] = tmp.mat[4][3] = tmp.mat[5][5] = 1; while(n) { if(n & 1)ans = ans * tmp; tmp = tmp * tmp; n >>= 1; } int res = 0; for(int i = 0; i < 4; i++) res = (res + ans.mat[5][i] % MOD) % MOD; res = res + (ans.mat[5][5] * 2) % MOD; return res % MOD; } int main() { int x,n,y; k = 6; MOD = 10007; while(~scanf("%d%d%d",&n,&x,&y)) { printf("%d\n",quick(n,x,y)); } return 0; }