HDU 5863 cjj's string game(矩阵快速幂)

$这题是典型的矩阵快速幂的数据范围$
$可以列出状态转移，dp[i][j][1]=dp[i-1][j-1][1]*k，i表示到第i个字符，j表示到第i个为止，有几个相同，1表示已经出现了m个相同，dp[i][j][0]表示还没出现m个相同的$
$然后因为i很大，后两维一共就2*m+2，所以可以变成矩阵快速幂，(2*m+2)*(2*m+2)的矩阵，表示的状态就是上面的dp的状态$

---

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           200005
#define   MAXN          1000005
#define   maxnode       205
#define   sigma_size    26
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-4;
const LL    mod    = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

struct Matrix{
    int n;
    LL maze[maxnode][maxnode];

    void init(int n){
        this->n = n;
        mem(maze,0);
    }
    Matrix operator * (Matrix &rhs){
        Matrix m;
        m.init(n);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                for(int k=0;k<n;k++)
                    m.maze[i][j] = (m.maze[i][j] + maze[i][k] * rhs.maze[k][j])%mod;
                    //mod可以在二重循环的地方模
        return m;
    }
};
int k;
LL qpow(Matrix a,int n){
    Matrix ans;
    ans.init(a.n);
    for(int i=0;i<ans.n;i++) ans.maze[i][i] = 1;
    while(n){
        if(n&1) ans = ans * a;
        a = a*a;
        n >>= 1;
    }
    LL sum=0;
    for(int i=0;i<(ans.n+1)/2+1;i++){
        sum+=(ans.maze[i][ans.n-1]*k*(k-1)+ans.maze[i][ans.n-2]*k)%mod;
        sum%=mod;
    }
    return sum;
}

int main(){
    int t;
    cin>>t;
    while(t--){
        int n,m;
        scanf("%d%d%d",&n,&m,&k);
        Matrix A;
        A.init(2*m+2);
        for(int i=0;i<2*m+2;i++){
            if(i<m) A.maze[i][i+1]=k;
            else if(i==m){
                for(int j=0;j<=i+1;j++){
                    A.maze[i][j]=k*(k-1);
                }
            }
            else if(i<2*m+1){
                A.maze[i][i+1]=k;
            }
            else{
                for(int j=m+2;j<=i;j++){
                    A.maze[i][j]=k*(k-1);
                }
            }
        }
        cout<<qpow(A,n-1)<<endl;
    }
    return 0;
}