POJ 3150 矩阵快速幂

抄的代码。

看不出矩阵快速幂，2B。。

注意到这是种变换，每次都是n个元素在变来变去，数据范围很大，这应该反应到时矩阵

根据大神指示，用O（n）复杂度存储空间，由于 A[i][j]=A[i-1][j-1] 

O(log(k) * n ^ 2)

据说可以用傅里叶变换，更快，没试

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <set>
#include <algorithm>
#include <vector>
#include <string>
using namespace std;
int n,m,d,k;
int init[505],temp[505];
void mul(int *a,int *b)
{
    int c[505]={0};
    for(int i=0;i<n;++i)
        for(int j=0;j<n;++j)
    {
        c[i]+=a[j]*b[ i>=j ? (i-j) : (n+i-j)];
        c[i]%=m;
    }
    for(int i=0;i<n;++i)
        a[i]=c[i];
}
int main ()
{

    while(scanf("%d%d%d%d",&n,&m,&d,&k)!=EOF)
    {
        for(int i=0;i<n;++i)
            scanf("%d",&init[i]);
        temp[0]=1;
        for(int i=1;i<=d;++i)
            temp[i]=temp[n-i]=1;
        while(k)
        {
            if(k&1)
                mul(init,temp);
            k>>=1;
            mul(temp,temp);
        }
        for(int i=0;i<n;++i)
        {
            printf("%d",init[i]);
            if(i<n-1)
                printf(" ");
            else printf("\n");
        }
    }
    return 0;
}