矩阵快速幂（233矩阵 HDU 5015）

最新推荐文章于 2020-04-07 15:20:48 发布

冷月残星

最新推荐文章于 2020-04-07 15:20:48 发布

阅读量328

点赞数

分类专栏：数学题

数学题专栏收录该内容

55 篇文章 0 订阅

订阅专栏

http://blog.csdn.net/rowanhaoa/article/details/39343769

我的代码

#include<stdio.h>
#include<string.h>
#define mod 10000007
using namespace std;

struct Mat
{
    int n,m;
    long long MAP[15][15];
    Mat(){memset(MAP,0,sizeof(MAP));}
    void init()
    {
        for(int i=0;i<n;i++) MAP[i][i]=1;
    }
    Mat operator * (const Mat& rhs) const
    {
        Mat ret;
        ret.n=n;
        ret.m=rhs.m;
        for(int i=0;i<ret.n;i++)
            for(int j=0;j<ret.m;j++)
                for(int k=0;k<m;k++)
                    ret.MAP[i][j]=(ret.MAP[i][j]+(MAP[i][k]*rhs.MAP[k][j])%mod)%mod;
        return ret;
    }
};

Mat operator ^ (Mat x,int n)
{
    Mat ret;
    ret.n=ret.m=x.n;
    ret.init();
    while(n)
    {
        if(n&1) ret=ret*x;
        x=x*x;
        n>>=1;
    }
    return ret;
}

int main()
{
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        Mat A;
        A.n=n+2;
        A.m=1;
        for(int i=1;i<=n;i++) scanf("%I64d",&A.MAP[i][0]);
        A.MAP[0][0]=23;
        A.MAP[n+1][0]=3;
        Mat B;
        B.n=B.m=n+2;
        for(int i=0;i<n+1;i++)
        {
            B.MAP[i][0]=10;
            B.MAP[i][n+1]=1;
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=i;j++)
                B.MAP[i][j]=1;
        B.MAP[n+1][n+1]=1;
        Mat C=(B^m)*A;
        printf("%I64d\n",C.MAP[n][0]);
    }
    return 0;
}