Hdu 1757 A Simple Math Problem//矩阵乘法

最新推荐文章于 2022-10-10 22:20:10 发布

mr_zys

最新推荐文章于 2022-10-10 22:20:10 发布

阅读量644

点赞数

分类专栏： Hdu解题报告文章标签：数学矩阵乘法

本文链接：https://blog.csdn.net/mr_zys/article/details/9405603

版权

Hdu解题报告专栏收录该内容

21 篇文章 0 订阅

订阅专栏

唉，是一道水题，但是错把b也为a，狂wa。。。

还有每次取模，否则，溢出。

下面是代码：

#include<stdio.h>
#include<string.h>
#define maxn  15
struct node
{
    int m[maxn][maxn];
    void init()
    {
        memset(m,0,sizeof(m));
    }
} A,B;
int k,m;
void init()
{
    A.init();
    B.init();
    for(int i = 1,j = 9; i < 10; i++,j--)
    {
        A.m[i][i-1] = 1;
        B.m[i-1][0] = j;
    }
    B.m[9][0] = 0;
}
node matrix_mul(node a,node b)
{
    node temp;
    temp.init();
    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++) if(b.m[k][j])
                temp.m[i][j] = (temp.m[i][j] + a.m[i][k] * b.m[k][j])%m;
        }
    }
    return temp;
}
node matrix_pow(int t)
{
    node temp;
    temp.init();
    for(int i = 0; i < 10; i++)
        temp.m[i][i] = 1;
    while(t)
    {
        if(t%2) temp = matrix_mul(temp,A);
        t = t/2;
        A = matrix_mul(A,A);
    }
    return temp;
}
int main()
{
    while(scanf("%d%d",&k,&m) != EOF)
    {
        init();
        for(int i = 0; i < 10; i++)
        {
            scanf("%d",&A.m[0][i]);
        }
        if(k < 10) printf("%d\n",k%m);
        else
        {
            node temp = matrix_pow(k-9);
            int ans = 0;
            for(int i = 0; i < 10; i++)
                ans = (ans+temp.m[0][i] * B.m[i][0])%m;
            printf("%d\n",ans);
        }
    }
    return 0;
}