码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 1757 A Simple Math Problem (矩阵快速幂)

时间:2014-06-16 12:06:07      阅读:231      评论:0      收藏:0      [点我收藏+]

标签:class   blog   code   http   tar   get   

这一题构造的矩阵的方法相同。

需要注意的是,题目中a0~a9 与矩阵相乘的顺序。


#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define N 10

using namespace std;
int mod;
typedef long long LL;

struct matrix
{
    LL a[10][10];
}origin;


int n=10,m;

matrix multiply(matrix x,matrix y)
{
    matrix temp;
    memset(temp.a,0,sizeof(temp.a));
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<n;k++)
            {
                temp.a[i][j]+=x.a[i][k]*y.a[k][j];
                temp.a[i][j]=(temp.a[i][j])%mod;
            }
        }
    }
    return temp;
}

matrix matmod(matrix A,int k)
{
    matrix res;

    memset(res.a,0,sizeof res.a);
    for(int i=0;i<n;i++)res.a[i][i]=1;

    while(k)
    {
        if(k&1)
        res=multiply(res,A);
        k>>=1;
        A=multiply(A,A);
    }
    return res;
}

void print(matrix x)
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        cout<<" "<<x.a[i][j];
        puts("");
    }
    printf("---------------\n");
}

int main()
{
    int k;
    while(scanf("%d%d",&k,&mod)!=EOF)
    {
        memset(origin.a,0,sizeof origin.a);
        for(int i=0;i<10;i++)
        {
            origin.a[i][0]=i;
        }
        //print(origin);
        matrix res;
        memset(res.a,0,sizeof res.a);

        for(int i=0;i<9;i++)res.a[i][i+1]=1;
        for(int i=9;i>=0;i--)scanf("%d",&res.a[9][i]);
        //print(res);
        res=matmod(res,k);

        matrix ans = multiply(res,origin);

        printf("%d\n",ans.a[0][0]);
    }
    return 0;
}


hdu 1757 A Simple Math Problem (矩阵快速幂),布布扣,bubuko.com

hdu 1757 A Simple Math Problem (矩阵快速幂)

标签:class   blog   code   http   tar   get   

原文地址:http://blog.csdn.net/u010709592/article/details/30144433

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!