题目来源:https://www.nowcoder.com/question/next?
思路
看这道题第一反应就是打表找规律,计算循环节,然后超时。
然后看了下讨论区,思路是矩阵快速幂,然后就根据条件推出了中间矩阵,
比样例的中间矩阵就是:
1 0 1
1 1 0
0 0 1,
然后就是一发快速幂,代码比较丑。。
代码
#include<set>
#include<cmath>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int INF=1e9+7;
typedef long long LL;
int n,k;
struct mat
{
int a[60][60];
};
mat operator*(mat r,mat t)
{
mat s;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
s.a[i][j]=0;
for(int q=1;q<=n;q++)
{
s.a[i][j]+=r.a[i][q]*t.a[q][j];
s.a[i][j]%=100;
}
}
}
return s;
}
void solve(mat A,mat B)
{
mat tmp;
for(int i=1;i<=n;i++)
tmp.a[i][i]=1;
while(k)
{
if(k&1)
{
tmp=tmp*B;
}
B=B*B;
k>>=1;
}
A=A*tmp;
printf("%d",A.a[1][1]);
for(int i=2;i<=n;i++)
{
printf(" %d",A.a[1][i]);
}
printf("\n");
}
void print(mat tmp)
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
printf("%d ",tmp.a[i][j]);
}
printf("\n");
}
}
int main()
{
scanf("%d%d",&n,&k);
mat A,B;
memset(A.a,0,sizeof(A.a));
for(int i=1;i<=n;i++)
{
scanf("%d",&A.a[1][i]);
}
memset(B.a,0,sizeof(B.a));
B.a[1][1]=B.a[1][n]=1;
for(int i=2;i<=n;i++)
{
B.a[i][i-1]=B.a[i][i]=1;
}
solve(A,B);
}