hdu 1757 A Simple Math Problem(矩阵优化递推)

最新推荐文章于 2021-04-02 16:24:12 发布

原创最新推荐文章于 2021-04-02 16:24:12 发布 · 618 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#矩阵优化递推

递推/矩阵专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种使用矩阵快速幂求解特定数学问题的方法，通过高效算法减少了计算复杂度，特别适用于大规模数据处理场景。

hdu 1757 A Simple Math Problem

赤果果的矩阵

#include<stdio.h>
#include<string.h>
struct matrix
{
    int m[10][10];
};
int mod;
matrix multi(const matrix &a,const matrix &b)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<10;i++)
        for(int k=0;k<10;k++)
            if(a.m[i][k])
            {
                for(int j=0;j<10;j++)
                {
                    ans.m[i][j]+=a.m[i][k]*b.m[k][j];
                    ans.m[i][j]%=mod;
                }
            }
    return ans;
}
matrix pow(matrix a,int k)
{
    matrix ans;
    memset(&ans,0,sizeof(ans));
    for(int i=0;i<10;i++) ans.m[i][i]=1;
    while(k)
    {
        if(k&1) ans=multi(ans,a);
        a=multi(a,a);
        k>>=1;
    }
    return ans;
}
int main()
{
    matrix a;
    int n;
    while(scanf("%d%d",&n,&mod)!=EOF)
    {
        memset(&a,0,sizeof(a));
        for(int i=0;i<10;i++)
            scanf("%d",&a.m[0][i]);
        
        if(n<10)
            printf("%d\n",n);
        else
        {
            for(int i=1;i<10;i++)
                a.m[i][i-1]=1;
            a=pow(a,n-9);
            int ans=0;
            for(int i=0;i<10;i++)
            {
                ans+=a.m[0][i]*(10-i-1);
                ans%=mod;
            }
            printf("%d\n",ans);
            
        }
    }
    return 0;
}