POJ 3150 循环矩阵的应用

思路：

1. 首先 先普及一个性质： 循环矩阵*循环矩阵=循环矩阵
   由于此题是距离小于d的都加上一个数。
   那么 构造矩阵的时候 我们发现 诶呦 这是个循环矩阵
   看看数据范围 n^2log(k)可以过。
   那就把这个矩阵改一改。
   因为这是个循环矩阵， 所以呢 只用保存一行就可以了。
   每回做乘法的时候只做第一行的乘法。
   for(i) for(j) temp[i]+=a[j]*b[(i+j)%n];
   就这么着 搞搞就能过了。

2. （好像可以用FFT？ 表示并不会）

// by SiriusRen
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
long long f[666],co[666],c[666];
int n,m,d,k;
void mul(LL *a,LL *b){
    memset(c,0,sizeof(c));
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            c[i]+=a[j]*b[(i+j)%n];
    for(int i=0;i<n;i++)b[i]=c[i]%m;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&d,&k);
    for(int i=0;i<n;i++)scanf("%lld",&f[i]);
    for(int i=0;i<=d;i++)co[i]=co[n-i]=1;
    while(k){
        if(k&1)mul(co,f);
        mul(co,co),k>>=1;
    }
    for(int i=0;i<n;i++)printf("%lld ",f[i]);
}

// by SiriusRen
#include <cstdio>
#define LL long long
#define f(Q,R) for(int Q=0;Q<R;Q++) 
using namespace std;
LL f[666],co[666],c[666];
int n,m,d,k;
void mul(LL *a,LL *b){
    f(i,n)c[i]=0;
    f(i,n)f(j,n)c[(i+j)%n]+=a[i]*b[j];
    f(i,n)b[i]=c[i]%m;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&d,&k);
    f(i,n)scanf("%lld",&f[i]);
    f(i,d+1)co[i]=co[n-i]=1;
    while(k){
        if(k&1)mul(co,f);
        mul(co,co),k>>=1;
    }
    f(i,n)printf("%lld ",f[i]);
}

Code length能进前三的存在哈哈哈

转载于:https://www.cnblogs.com/SiriusRen/p/6532384.html