Given a n × n matrix A and a positive integer k, find the sum S = A + A2 + A3 + … + Ak.
The input contains exactly one test case. The first line of input contains three positive integers n (n ≤ 30), k (k ≤ 109) and m (m < 104). Then follow n lines each containing n nonnegative integers below 32,768, giving A’s elements in row-major order.
Output the elements of S modulo m in the same way as A is given.
2 2 4 0 1 1 1
1 2 2 3
方法一:
构造矩阵
通过矩阵分块构造出矩阵S,从这三个式子可以看出右上角的那个矩阵减去单位矩阵就是我们要求的结果(构造矩阵的形式可能多样,最后解所处的位置可能不同)
注意一点就是,如果对角线上出现负数就加上取模的的那个数(毕竟由非负数构成的矩阵不可能出现负数,所以负数的出现只能是取模造成的)
#include<stdio.h>
#include<string.h>
typedef struct Matrix
{
int M[65][65];
Matrix()
{
memset(M,0,sizeof(M));
}
}Matrix;
Matrix M_multi(Matrix m1,Matrix m2,int n,int m)
{
Matrix m3;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
for(int k=0;k<n;k++)
{
m3.M[i][j]+=(m1.M[i][k]%m)*(m2.M[k][j]%m);
}
m3.M[i][j]%=m;
}
}
return m3;
}
int main()
{
int n,k,m;
Matrix t,s;
scanf("%d%d%d",&n,&k,&m);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&t.M[i][j]);
t.M[i][i+n]=t.M[i+n][i+n]=s.M[i][i]=s.M[i+n][i+n]=1;
}
}
k++;
while(k)
{
if(k&1)
{
s=M_multi(s,t,2*n,m);
}
t=M_multi(t,t,2*n,m);
k>>=1;
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
if(i==j)
s.M[i][j+n]--;
if(s.M[i][j+n]<0)
s.M[i][j+n]+=m;
printf("%d",s.M[i][j+n]%m);
if(j!=n-1)
putchar(' ');
else
putchar('\n');
}
}
return 0;
}
利用递归的思想,进行递归求解(E为单位矩阵)
如果指数为偶数:
指数为奇数:
递归边界为指数为1时,直接返回A
#include<stdio.h>
#include<string.h>
typedef struct Matrix
{
int M[35][35];
Matrix()
{
memset(M,0,sizeof(M));
}
}Matrix;
int n,k,m;
Matrix E;
Matrix M_add(Matrix m1,Matrix m2)
{
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
m1.M[i][j]+=m2.M[i][j];
m1.M[i][j]%=m;
}
}
return m1;
}
Matrix M_multi(Matrix m1,Matrix m2)
{
Matrix m3;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
for(int k=0;k<n;k++)
{
m3.M[i][j]+=(m1.M[i][k]%m)*(m2.M[k][j]%m);
}
m3.M[i][j]%=m;
}
}
return m3;
}
Matrix M_pow(Matrix t,int cur)
{
Matrix s=E;
while(cur)
{
if(cur&1)
{
s=M_multi(s,t);
}
t=M_multi(t,t);
cur>>=1;
}
return s;
}
Matrix M_sum(Matrix s,int cur)
{
if(cur==1)
return s;
Matrix tmp=M_sum(s,cur>>1); //前k/2项的和
if(cur&1)
return M_add(M_multi(tmp,M_add(M_pow(s,cur>>1),E)),M_pow(s,cur));
else
return M_multi(tmp,M_add(M_pow(s,cur>>1),E));
}
int main()
{
Matrix ans,s;
scanf("%d%d%d",&n,&k,&m);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&s.M[i][j]);
}
}
for(int i=0;i<n;i++)
E.M[i][i]=1;
ans=M_sum(s,k);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
printf("%d",ans.M[i][j]%m);
if(j!=n-1)
putchar(' ');
else
putchar('\n');
}
}
return 0;
}