HDU5015 233 Matrix (2014 ACM/ICPC Asia Regional Xi'an Online)

最新推荐文章于 2019-04-19 22:05:23 发布

猪猪奋斗记

最新推荐文章于 2019-04-19 22:05:23 发布

阅读量666

点赞数

分类专栏：数学

本文链接：https://blog.csdn.net/bigbigship/article/details/39313449

版权

数学专栏收录该内容

119 篇文章 0 订阅

订阅专栏

题意：给定一个矩阵的第一行第一列求 Anm

状态转移矩阵如下：

代码如下：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 12;

const int M = 10000007;

int n,m;

struct matrix
{
    long long m[N][N];
};

matrix I,A;

matrix multi(matrix a,matrix b)
{
    matrix c;
    for(int i=0;i<n+2;i++)
        for(int j=0;j<n+2;j++){
                c.m[i][j]=0;
            for(int k=0;k<n+2;k++)
                c.m[i][j]+=(a.m[i][k]   %M*b.m[k][j]%M)%M;
            c.m[i][j]%=M;
        }
    return c;
}

matrix pow(matrix A,int  k)
{
    matrix ans=I,p=A;
    while(k){
        if(k&1){
            ans=multi(ans,p);
            k--;
        }
        k>>=1;
        p=multi(p,p);
    }
    return ans;
}

void init()
{
    for(int i=0;i<n+2;i++){
        for(int j=0;j<n+2;j++)
            I.m[i][j]=(i==j);
    }
    for(int i=0;i<N;i++){
        for(int j=0;j<N;j++){
            if(j<=i)
                A.m[i][j]=1;
            else
                A.m[i][j]=0;
            if(j==1&&i)
                A.m[i][j]=10;
        }
    }
}

int a[12];

int main()
{
    while(~scanf("%d%d",&n,&m)){
        a[0]=3;a[1]=23;
        init();
        for(int i=0;i<n;i++)
            scanf("%d",&a[i+2]);
        matrix ans=pow(A,m);
        long long cnt=0;
        for(int i=0;i<n+2;i++)
            cnt=(cnt+ans.m[n+1][i]*a[i]%M)%M;
        printf("%I64d\n",cnt);
    }
    return 0;
}