1 #define maxn 15 2 3 int n, k; // n阶方阵的k次幂 4 5 struct matrix//定义一个结构体，方便传递值 6 { 7 int m[maxn][maxn]; 8 }; 9 10 /* 11 mod由全局定义，mod根据需要可以省去 12 */ 13 14 matrix mul(matrix a, matrix b) //矩阵求积, 矩阵乘法 15 { 16 matrix ans; 17 for(int i = 1; i <= n; i++) 18 { 19 for(int j = 1; j <= n; j++) 20 { 21 ans.m[i][j] = 0; 22 for(int k = 1; k <= n; k++) 23 { 24 ans.m[i][j] += (a.m[i][k] * b.m[k][j]) % mod; 25 ans.m[i][j] %= mod; // 这里一定要再次取模 26 } 27 } 28 } 29 return ans; 30 } 31 32 matrix quick_pow(matrix a, int b) //矩阵快速幂 33 { 34 matrix ans; 35 for(int i = 1; i <= n; i++) 36 { 37 for(int j = 1; j <= n; j++) 38 { 39 if(i == j) 40 ans.m[i][j] = 1; 41 else 42 ans.m[i][j] = 0;//这里要初始化为单位矩阵，类比普通快速幂这里初始化为1 43 } 44 } 45 while(b != 0)//方法与普通快速幂相同，只有乘法的实现不同 46 { 47 if(b % 2 == 1) 48 ans = mul(a, ans); 49 a = mul(a, a); 50 b /= 2; 51 } 52 53 return ans; 54 }