vijos1067 Warcraft III 守望者的烦恼(矩阵倍增)

来源:互联网 发布:手机淘宝如何退保证金 编辑:程序博客网 时间:2024/05/22 10:35

f[i]=f[i-1]+f[i-2]+…+f[i-k].根据递推式构造出转移矩阵trans,trans.mat[1][n]表示f[i]。初始时f[0]=1.这便是矩阵倍增的真正用处了。。。

#include <cstdio>#include <cstring>#define N 12#define mod 7777777#define ll long longint n,m;inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}struct Matrix{    int mat[N][N];    Matrix(bool t){        memset(mat,0,sizeof(mat));        if(t) for(int i=1;i<=n;++i) mat[i][i]=1;    }    Matrix operator*(Matrix b){        Matrix res(0);        for(int i=1;i<=n;++i)            for(int j=1;j<=n;++j)                for(int k=1;k<=n;++k)                    res.mat[i][j]=(res.mat[i][j]+(ll)mat[i][k]*b.mat[k][j])%mod;        return res;    }    Matrix operator^(int k){        Matrix res(1),base(0);        memcpy(base.mat,mat,sizeof(mat));        for(;k;k>>=1,base=base*base)            if(k&1) res=res*base;        return res;    }}ans(0),trans(0);int main(){//  freopen("a.in","r",stdin);    n=read();m=read();    for(int i=1;i<n;++i) trans.mat[i+1][i]=1;    for(int i=1;i<=n;++i) trans.mat[i][n]=1;    ans.mat[1][n]=1;    ans=ans*(trans^m);    printf("%d\n",ans.mat[1][n]);    return 0;}
原创粉丝点击