标签：
K(3<=K<=10^9)个人互相传球，某人接球后立即传给别人。假定初始状态球在甲手中，并将甲发球作为第一次传球过程。求经过N(N<=10^9)次传球后，球又回到甲手中的传球方案数，输出这个数模10^9+7后的结果。
第一行是一个整数T(T<=20000)，表示测试数据的组数。
接下来T行，每行输入两个数N,K(3<=K<=10^9,1<= N<=10^9)。
输出T行，每行输出一组N,K对应方案数模10^9+7后的结果。
2 3 3 3 4
2 6
第一组样例，N=3,K=3，三个人传三次的传球方式是：
1. A->B->C->A
2. A->C->B->A
1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 using namespace std; 6 #define LL long long 7 #define mod 1000000007 8 struct matrix 9 { 10 LL mat[2][2]; 11 }; 12 13 matrix multiply(matrix a,matrix b) 14 { 15 matrix c; 16 memset(c.mat,0,sizeof(c.mat)); 17 for(int i=0;i<2;i++) 18 { 19 for(int j=0;j<2;j++) 20 { 21 if(a.mat[i][j]==0)continue; 22 for(int k=0;k<2;k++) 23 { 24 if(b.mat[j][k]==0)continue; 25 c.mat[i][k]+=a.mat[i][j]*b.mat[j][k]%mod; 26 // c.mat[i][k]%=mod; 27 if(c.mat[i][k]>mod) c.mat[i][k]-=mod;//果然这里超了。。。 28 else if(c.mat[i][k]<0) c.mat[i][k]+=mod; 29 } 30 } 31 } 32 return c; 33 } 34 35 matrix quicklymod(matrix a,LL n) 36 { 37 matrix res; 38 memset(res.mat,0,sizeof(res.mat)); 39 for(int i=0;i<2;i++) res.mat[i][i]=1; 40 while(n) 41 { 42 if(n&1) 43 res=multiply(a,res); 44 a=multiply(a,a); 45 n>>=1; 46 } 47 return res; 48 } 49 50 int main() 51 { 52 LL N,K; 53 int T; 54 scanf("%d",&T); 55 while(T--) 56 { 57 scanf("%lld%lld",&N,&K); 58 if(N==1){printf("0\n");continue;} 59 //if(N==2){printf("%lld\n",K-1);continue;} 60 61 matrix ans; 62 ans.mat[0][0]=K-1; 63 ans.mat[0][1]=0; 64 ans.mat[1][0]=K-1; 65 ans.mat[1][1]=-1; 66 67 // ans=quicklymod(ans,N-2); 68 // LL res=(((K-1)%mod)*(ans.mat[1][0]+ans.mat[1][1])%mod)%mod; 69 // printf("%lld\n",res); 70 ans=quicklymod(ans,N-1); 71 printf("%lld\n",ans.mat[1][0]); 72 } 73 return 0; 74 }
1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 using namespace std; 6 #define LL long long 7 #define mod 1000000007 8 struct matrix 9 { 10 LL mat[2][2]; 11 }; 12 13 matrix multiply(matrix a,matrix b) 14 { 15 matrix c; 16 memset(c.mat,0,sizeof(c.mat)); 17 for(int i=0;i<2;i++) 18 { 19 for(int j=0;j<2;j++) 20 { 21 if(a.mat[i][j]==0)continue; 22 for(int k=0;k<2;k++) 23 { 24 if(b.mat[j][k]==0)continue; 25 c.mat[i][k]=(c.mat[i][k]+a.mat[i][j]*b.mat[j][k])%mod; 26 } 27 } 28 } 29 return c; 30 } 31 32 matrix quicklymod(matrix a,LL n) 33 { 34 matrix res; 35 memset(res.mat,0,sizeof(res.mat)); 36 for(int i=0;i<2;i++) res.mat[i][i]=1; 37 while(n) 38 { 39 if(n&1) 40 res=multiply(a,res); 41 a=multiply(a,a); 42 n>>=1; 43 } 44 return res; 45 } 46 47 int main() 48 { 49 LL N,K; 50 int T; 51 scanf("%d",&T); 52 while(T--) 53 { 54 scanf("%lld%lld",&N,&K); 55 if(N==1){printf("0\n");continue;} 56 // if(N==2){printf("%lld\n",K-1);continue;} 57 58 matrix ans; 59 ans.mat[0][0]=0; 60 ans.mat[0][1]=K-1; 61 ans.mat[1][0]=1; 62 ans.mat[1][1]=K-2; 63 64 ans=quicklymod(ans,N-1); 65 // LL res=((K-1)*(ans.mat[1][0]+ans.mat[1][1])%mod)%mod; 66 // printf("%lld\n",res); 67 printf("%lld\n",ans.mat[0][1]); 68 } 69 return 0; 70 }
1 #include<stdio.h> 2 #include<algorithm> 3 using namespace std; 4 long long pow(long long n,long long k) 5 { 6 long long res = 1; 7 while (k) 8 { 9 if (k&1) res = res*n%1000000007; 10 n = n*n%1000000007; 11 k >>= 1; 12 } 13 return res; 14 } 15 long long cal(long long n,long long k) 16 { 17 long long res = pow(k-1,n); 18 if(res && n & 1) 19 res = 1000000007 - res; 20 res += (k-1); 21 if (res >= 1000000007) res -= 1000000007; 22 res = res * pow(k,1000000005)%1000000007; 23 if(res && n & 1) 24 res = 1000000007 - res; 25 return res; 26 } 27 int main() 28 { 29 int _; 30 long long N,K; 31 scanf("%d",&_); 32 while (_--) 33 { 34 scanf("%lld %lld",&N,&K); 35 printf("%lld\n",cal(N,K)); 36 } 37 return 0; 38 }
标签：
原文地址：http://www.cnblogs.com/yuyixingkong/p/4491495.html
