Problem 1013 - 时间工厂
/*Time Limit: 1000MS Memory Limit: 65536KB Difficulty:
Total Submit:
237 Accepted: 38 Special Judge:
No
Description
zyf总是有很多奇异的想法,他最近常常幻想着以后能开这么一个工厂,可以把前三天里生产出来的东西拿到今天来拼在一起作为今天生产的东西。假如前三天生产出来的产品数分别是x,y,z,那么今天就能生产出x+y+z个。这样一来只要前三天的投入,接下来的工厂每一天都是0成本运作,但产品数却在极速增加,相当暴利。
当然,为了防止地球被破坏,为了维护世界的和平,zyf是不会让工厂每天生产出来的东西超过1000000006个的,如果超过了,就不停减去1000000007,减到不超过为止。
现在zyf想知道如果第一、二、三天分别生产a,b,c个产品的话,第n天会生产出多少产品呢?
Input
输入数据的第一行case数。
接下来每一行都有四个数字a,b,c,n(1<=a,b,c,n<=10^9),意义如上文.
Output
对于每个case输出一行,第n天生产的产品数。
Sample
Input
2
1 2 3 4
1 1 1 5
Sample Output
6
5
*/
矩阵快速幂。首先构造矩阵A=[0 0 1;1 0 1;0 1 1];X=[a b c] (a,b,c为前三个元素)
x*A=[b c a+b+c];所以可知所求结果可应用矩阵的运算。但是数据太大,所以采用快速幂的方法求。下面是AC代码。
1 #include<string.h> 2 #include<stdlib.h> 3 #define INF 1000000007 4 #define MAX 3 5 typedef struct Mar 6 { 7 long long m[MAX][MAX]; //long long 我原来写的int让我调了好半天。。。 8 }Mar; 9 Mar A,X; 10 int INIT() 11 { 12 int i,j; 13 A.m[0][0]=0;A.m[0][1]=0;A.m[0][2]=1; 14 A.m[1][0]=1;A.m[1][1]=0;A.m[1][2]=1; 15 A.m[2][0]=0;A.m[2][1]=1;A.m[2][2]=1; 16 for(i=0;i<MAX;i++) 17 for(j=0;j<MAX;j++) 18 X.m[i][j]=0; 19 } 20 Mar mul(Mar x,Mar y) 21 { 22 int i,j,k; 23 Mar temp; 24 memset(temp.m,0,sizeof(temp.m)); 25 temp.m[0][0]=(x.m[0][0]*y.m[0][0]%INF+x.m[0][1]*y.m[1][0]%INF+x.m[0][2]*y.m[2][0]%INF)%INF; 26 temp.m[0][1]=(x.m[0][0]*y.m[0][1]%INF+x.m[0][1]*y.m[1][1]%INF+x.m[0][2]*y.m[2][1]%INF)%INF; 27 temp.m[0][2]=(x.m[0][0]*y.m[0][2]%INF+x.m[0][1]*y.m[1][2]%INF+x.m[0][2]*y.m[2][2]%INF)%INF; 28 temp.m[1][0]=(x.m[1][0]*y.m[0][0]%INF+x.m[1][1]*y.m[1][0]%INF+x.m[1][2]*y.m[2][0]%INF)%INF; 29 temp.m[1][1]=(x.m[1][0]*y.m[0][1]%INF+x.m[1][1]*y.m[1][1]%INF+x.m[1][2]*y.m[2][1]%INF)%INF; 30 temp.m[1][2]=(x.m[1][0]*y.m[0][2]%INF+x.m[1][1]*y.m[1][2]%INF+x.m[1][2]*y.m[2][2]%INF)%INF; 31 temp.m[2][0]=(x.m[2][0]*y.m[0][0]%INF+x.m[2][1]*y.m[1][0]%INF+x.m[2][2]*y.m[2][0]%INF)%INF; 32 temp.m[2][1]=(x.m[2][0]*y.m[0][1]%INF+x.m[2][1]*y.m[1][1]%INF+x.m[2][2]*y.m[2][1]%INF)%INF; 33 temp.m[2][2]=(x.m[2][0]*y.m[0][2]%INF+x.m[2][1]*y.m[1][2]%INF+x.m[2][2]*y.m[2][2]%INF)%INF; 34 return temp; 35 } 36 37 void quickpow(int n) 38 { 39 int i,j; 40 while(n) 41 { 42 if (n&1) 43 { 44 X=mul(X,A); 45 } 46 n>>=1; 47 A=mul(A,A); 48 } 49 printf("%lld\n",X.m[0][2]); 50 51 } 52 int main() 53 { 54 int i,j,T; 55 long long n; 56 57 scanf("%d",&T); 58 while(T--) 59 { 60 INIT(); 61 scanf("%d%d%d%lld",&X.m[0][0],&X.m[0][1],&X.m[0][2],&n); 62 if(n==1) {printf("%lld\n",X.m[0][0]); continue;} 63 if(n==2) {printf("%lld\n",X.m[0][1]); continue;} 64 if(n==3) {printf("%lld\n",X.m[0][2]); continue;} 65 quickpow(n-3); 66 } 67 return 0; 68 }