lightoj 1096【矩阵快速幂(作为以后的模板)】

基础矩阵快速幂何必看题解

#include <bits/stdc++.h>
using namespace std;

/*
0 1 2 3 4 5 6 7
0 0 0

*/
const int mod=10007;
struct asd{
int num[4][4];
};

asd mul(asd a,asd b)
{
asd ans;
memset(ans.num,0,sizeof(ans.num));
for(int i=0;i<4;i++)
for(int j=0;j<4;j++)
for(int k=0;k<4;k++)
ans.num[i][j]=(ans.num[i][j]+a.num[i][k]*b.num[k][j]%mod)%mod;
return ans;
}

asd quickmul(int g,asd x)
{
asd ans;
for(int i=0;i<4;i++)
for(int j=0;j<4;j++)
{
if(i==j)
ans.num[i][j]=1;
else
ans.num[i][j]=0;
}
while(g)
{
if(g%2)
ans=mul(ans,x);
x=mul(x,x);
g>>=1;
}
return ans;
}

int main()
{
int T,cas=1;
scanf("%d",&T);
while(T--)
{
int a,b,c,n;
scanf("%d%d%d%d",&n ,&a ,&b ,&c);
printf("Case %d: ",cas++);
if(n<=2)
{
puts("0");
continue;
}
asd tmp;
tmp.num[0][0]=a;tmp.num[0][1]=0;tmp.num[0][2]=b;tmp.num[0][3]=c;
tmp.num[1][0]=1;tmp.num[1][1]=0;tmp.num[1][2]=0;tmp.num[1][3]=0;
tmp.num[2][0]=0;tmp.num[2][1]=1;tmp.num[2][2]=0;tmp.num[2][3]=0;
tmp.num[3][0]=0;tmp.num[3][1]=0;tmp.num[3][2]=0;tmp.num[3][3]=1;

asd ans=quickmul(n-2,tmp);
printf("%d\n",ans.num[0][3]);
}
return 0;
}