(我想写的和这位博主一样,就转载了,表示这是一道好题,对矩阵乘法的优化有所启示)
【题意】
一个n个位置的环,每经过一次置换,每个位置的值会变成上一次周围d范围之内的和,求mod m后置换过k次后每个位置的值
很容易可以看出是矩阵乘法
n<=500
如果构建n*n的矩阵,矩阵乘法复杂度为n^3,那么总体复杂度为n^3log k 是不能接受的
但是可以发现一个矩阵的性质
因为是求d范围之内之和,所以矩阵每一行基本一样,只是错了一位而已
所以求矩阵的时候只需要求一行就够了,复杂度降到n^2,便可以通过了。
(我想写的和这位博主一样,就转载了,表示这是一道好题,对矩阵乘法的优化有所启示)
(代码是自己写的)
#include <iostream>
using namespace std;
const int maxn=505;
long long a[maxn],ans[maxn],c[maxn];
int n,m,d,k,i,j;
void mul(long long a[],long long b[])
{
long long c[maxn]={0};
int i,j;
for (i=0;i<n;i++)
for (j=0;j<n;j++)
c[i]+=a[j]*b[(i>=j)?i-j:i-j+n];
for (i=0;i<n;i++)
a[i]=c[i]%m;
}
int main()
{
freopen("pin.txt","r",stdin);
freopen("pou.txt","w",stdout);
cin >> n >> m >> d >> k;
for (i=0;i<n;i++)
cin >> c[i];
for (a[0]=i=1;i<=d;i++)
a[i]=a[n-i]=1;
ans[0]=1;
while (k)
{
if (k&1)
mul(ans,a);
mul(a,a);
k>>=1;
}
for (i=0;i<n;i++)
{
long long ss=0;
for (j=0;j<n;j++)
ss=(ss+ans[(j>=i)?j-i:j-i+n]*c[j])%m;
cout << ss;
if (i!=n-1) cout << " ";
else cout << endl;
}
return 0;
}