挑戰算法程序設計的例題
描述見問題出處
AC代碼：
#include <iostream>
#include <cstdio>
#include <string.h>
#include <queue>
#include <vector>
#include <cstdlib>
using namespace std;
typedef vector<vector<int> >mat;
typedef vector<int> vec;
const int mod=1e4+7;
mat mul(mat a,mat b){
mat c(a.size(),vec(b[0].size()));
for(int i=0;i<a.size();i++)
for(int j=0;j<b[0].size();j++)
for(int k=0;k<b.size();k++)
c[i][j]=((long long)c[i][j]+((long long)a[i][k]*b[k][j])%mod)%mod;
return c;
}
mat q_m_pow(mat a,int m){
if(m==1)return a;
mat c=q_m_pow(a,m/2);
if(m%2)
return mul(a,mul(c,c));
else
return mul(c,c);
}
int main()
{
int n,t;
cin>>t;
mat a(3,vector<int>(3));
while(t--){
cin>>n;
a[0][0]=2;a[0][1]=1;a[0][2]=0;
a[1][0]=2;a[1][1]=2;a[1][2]=2;
a[2][0]=0;a[2][1]=1;a[2][2]=2;
a=q_m_pow(a,n);
cout<<a[0][0]<<endl;
}
return 0;
}