题目大意:细胞上的数字会随着周围数字的和而变化,求第k轮之后数字的情况。
解题报告:很容易可以想到矩阵乘法。用快速幂的确可以很快的计算出k次方来。
可是n最大有500,矩阵乘法的复杂度是n^3,普通算法一定会超时。
这时就要看矩阵的性质了。仔细观察可以发现矩阵的每一行都是对称的,而且下一行左移一位就是上一行。这种矩阵叫循环矩阵。
循环矩阵的性质:循环矩阵A,与矩阵B,A*B仍然是循环矩阵。
根据性质,我们做矩阵乘法时可以只计算第一行。当然,因为第一行已经包含整个矩阵的信息了,所以我们可以只记录第一行。
用快速幂可以很快的计算出来答案。另外每次计算完之后取模会快很多。POJ上下面的代码是500MS。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long Array[555];
int mod;
int n;
Array a, b, c;
void mul(Array& a, Array& b)
{
memset(c,0,sizeof(c));
for(int i=0;i<n;i++)
for(int k=0;k<n;k++)
c[i]+=a[k]*b[k>=i?(k-i):(n+k-i)];
for(int i=0;i<n;i++)
a[i]=c[i]%mod;
}
void powArray(Array& a, int b)
{
Array res={1};
while(b)
{
if(b&1)
mul(res, a);
mul(a, a);
b>>=1;
}
memcpy(a, res, sizeof(res));
}
int main()
{
int d, k;
while(~scanf("%d%d%d%d", &n, &mod, &d, &k))
{
memset(a, 0, sizeof(a));
for(int i=0;i<=d;i++)
a[i]=a[(n-i)%n]=1;
powArray(a, k);
for(int i=0;i<n;i++)
scanf("%lld", b+i);
mul(b, a);
printf("%d", b[0]);
for(int i=1;i<n;i++)
printf(" %lld", b[i]);
puts("");
}
}
今年的杭州邀请赛的第一题也是这种类型,当时没做出来,因为连矩阵乘法都看不出来。今天一口气把这题也A了。
题目链接:HDU 4576
这道题很有意思……关键是细节。解题报告:解题报告