矩阵的题就是这么伤脑筋啊~~ sad……
题目大意:
一个环上有n个数,定义一种操作将它和它距离小于d的数加和再模m。每次操作刷新所有数。问k次之后都将变成什么数?
解题思路:
矩阵快速幂加速递推。
按照正常思路第i次操作是基于第i-1次操作完成的,也就是说要完成第i次操作需要先完成第i-1次。
但是用于矩阵之后可以直接推出第i次与第一次之间是什么关系。
这个矩阵是可以通过矩阵快速幂得出的。取模也是顺带的~
如果你注意关系矩阵的建立的话,你会发现这么一个规律。对于d=1来说:
b^1 =
[1, 1, 0, 0, 1]
[1, 1, 1, 0, 0]
[0, 1, 1, 1, 0]
[0, 0, 1, 1, 1]
[1, 0, 0, 1, 1]
b^2 =
[3, 2, 1, 1, 2]
[2, 3, 2, 1, 1]
[1, 2, 3, 2, 1]
[1, 1, 2, 3, 2]
[2, 1, 1, 2, 3]
b^3 =
[7, 6, 4, 4, 6]
[6, 7, 6, 4, 4]
[4, 6, 7, 6, 4]
[4, 4, 6, 7, 6]
[6, 4, 4, 6, 7]
b^4 =
[19, 17, 14, 14, 17]
[17, 19, 17, 14, 14]
[14, 17, 19, 17, 14]
[14, 14, 17, 19, 17]
[17, 14, 14, 17, 19]
也就是说我们只需要存第一行就行了。每次矩阵乘法也只需要得出第一行就行了。
下面是代码:
#include <set>
#include <map>
#include <queue>
#include <math.h>
#include <vector>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
#define eps 1e-7
#define pi acos(-1.0)
#define inf 107374182
#define inf64 1152921504606846976
#define lc l,m,tr<<1
#define rc m + 1,r,tr<<1|1
#define iabs(x) ((x) > 0 ? (x) : -(x))
#define clear1(A, X, SIZE) memset(A, X, sizeof(A[0]) * (SIZE))
#define clearall(A, X) memset(A, X, sizeof(A))
#define memcopy1(A , X, SIZE) memcpy(A , X ,sizeof(X[0])*(SIZE))
#define memcopyall(A, X) memcpy(A , X ,sizeof(X))
#define max( x, y ) ( ((x) > (y)) ? (x) : (y) )
#define min( x, y ) ( ((x) < (y)) ? (x) : (y) )
using namespace std;
struct node
{
long long matrix[505];
}a,temp;
int n,m,d,k;
node does(node a,node b)
{
node c;
clearall(c.matrix,0);
int p;
for(int i=0;i<n;i++)
{
p=(n-i)%n;
for(int j=0;j<n;j++)
{
c.matrix[i]+=a.matrix[j]*b.matrix[p++];
if(p==n)p=0;
}
c.matrix[i]%=m;
}
return c;
}
void mul(int k)
{
clearall(temp.matrix,0);
temp.matrix[0]=1;
while(k)
{
if(k&1)temp=does(temp,a);
a=does(a,a);
k>>=1;
}
}
int main()
{
long long num[505],ans[505];
int p;
while(scanf("%d%d%d%d",&n,&m,&d,&k)!=EOF)
{
clearall(ans,0);
for(int i=0;i<n;i++)
{
scanf("%lld",&num[i]);
}
for(int i=0;i<n;i++)
{
if(i<=d||n-i<=d)a.matrix[i]=1;
else a.matrix[i]=0;
}
mul(k);
for(int i=0;i<n;i++)
{
if(i!=0)printf(" ");
p=(n-i)%n;
for(int j=0;j<n;j++)
{
ans[i]+=temp.matrix[p++]*num[j];
if(p==n)p=0;
}
printf("%lld",ans[i]%m);
}
puts("");
}
return 0;
}