普通的快速幂
1 #include<stdio.h> 2 #define lll long long 3 lll ksm(int,int); 4 int x,n; 5 int main() 6 { 7 scanf("%d%d",&x,&n); 8 printf("%lld",ksm(x,n)); 9 return 0; 10 } 11 lll ksm(int x,int n) 12 { 13 lll ret = x; n--; 14 while(n) 15 { 16 if(n & 1) ret *= x; 17 x *= x; 18 n >>= 1; 19 } 20 return ret; 21 }
矩阵快速幂
1 #include<stdio.h> 2 #define maxn 105 3 #define mo 1000000007 4 #define lll long long 5 int n,m; 6 struct mat 7 { 8 lll ori[maxn][maxn]; 9 mat operator * (const mat &y) const 10 { 11 mat ret; 12 int i,j,k; 13 for(i=1;i<=n;i++) 14 for(j=1;j<=n;j++) 15 { 16 ret.ori[i][j] = 0; 17 for(k=1;k<=n;k++) 18 ret.ori[i][j] = (ret.ori[i][j]+ori[i][k]*y.ori[k][j])%mo; 19 } 20 return ret; 21 } 22 }; 23 mat A,ans; 24 mat ksm (int k,mat a) 25 { 26 mat ret = a; k--; 27 while(k) 28 { 29 if(k&1) ret = ret * a; 30 a = a*a; 31 k >>= 1; 32 } 33 return ret; 34 } 35 int main() 36 { 37 scanf("%d%d",&n,&m); 38 int i,j; 39 for(i=1;i<=n;i++) 40 for(j=1;j<=n;j++) 41 scanf("%d",&A.ori[i][j]); 42 ans = ksm(m,A); 43 for(i=1;i<=n;i++) 44 { 45 for(j=1;j<=n;j++) printf("%lld ",ans.ori[i][j]); 46 printf(" "); 47 } 48 return 0; 49 }