LightOj 1096 - nth Term （矩阵快速幂，简单）

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

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

matrix multiply(matrix x,matrix y)//矩阵乘法
{
matrix temp;
for(int i=0;i<num;i++)
{
for(int j=0;j<num;j++)
{
int ans=0;
for(int k=0;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)//n次矩阵快速幂
{
while(n)
{
if(n%2==1)//原来比赛时一直是这里被我不小心改成了0，怪不得案例一直都过不了
answ=multiply(origin,answ);
origin=multiply(origin,origin);
n/=2;
}
return answ;
}

int main()
{
int a,b,n,c,t;
scanf("%d",&t);
for(int id=1;id<=t;id++)
{
num=4;mod=10007;
scanf("%d%d%d%d",&n,&a,&b,&c);
matrix origin={1,0,0,0,
0,0,1,0,
0,0,0,1,
c,b,0,a};
matrix answ={1,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0};
answ=calc(origin,answ,n-2);
printf("Case %d: %d\n",id,answ.a[3][0]);
}
return 0;

}