挑戰演算法程式設計的例題

描述見問題出處

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;
}