hdu 1757 A Simple Math Problem （矩阵快速幂，简单）

题目

也是和LightOJ 1096 和LightOJ 1065 差不多的简单题目。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

int num,mod;
struct matrix
{
    int a[][];
};

matrix multiply(matrix x,matrix y)//矩阵乘法
{
       matrix temp;
       for(int i=;i<num;i++)
       {
               for(int j=;j<num;j++)
               {
                       int ans=;
                       for(int k=;k<num;k++)
                       {
                               ans+=((x.a[i][k]*y.a[k][j])%mod);
                       }
                       temp.a[i][j]=ans%mod;
               }
       }
       return temp;
}

matrix calc(matrix origin ,matrix answ,int n)//矩阵快速幂——answ*origin^n
{
     while(n)
     {
             if(n%==)//原来比赛时一直是这里被我不小心改成了0，怪不得案例一直都过不了
                    answ=multiply(origin,answ);
             origin=multiply(origin,origin);
             n/=;
     }
     return answ;
}

int main()
{
    int n;
    while(scanf("%d%d",&n,&mod)!=EOF)
    {
        matrix origin= {,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,,
                        ,,,,,,,,,};
        matrix answ={,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,,
                     ,,,,,,,,,};
        num=;
        for(int i=;i>=;i--)//要注意这里的a9~a0的输入顺序
            scanf("%d",&origin.a[][i]);

        answ=calc(origin,answ,n-);
        printf("%d\n",answ.a[][]%mod);
    }
    return ;
}