51nod 1033骨牌覆盖 V2

矩阵乘法题

就是状压一下一列的状态然后预处理出某两种状态的转移的方案数

然后构造一个矩阵a[i][j]表示i的状态到j的状态的条数

然后直接矩阵乘法就可以了

#include<algorithm>
#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
const int mod=1000000007;
typedef vector<long long> vec;
typedef vector<vec> mat;
// A*B
mat mul(mat& A, mat& B){
    mat C(A.size(),vec(B[0].size()));
    for(int i=0;i<(int)A.size();++i){
    	for(int j=0;j<(int)B[0].size();++j){
    		for(int k=0;k<(int)B.size();++k){
    			C[i][j]=(C[i][j]+A[i][k]*B[k][j])%mod;
			}
		}
	}
    return C;
}
// A^n
mat pow(mat A,int n){
    mat B(A.size(),vec(A.size()));
    for(int i=0;i<(int)A.size();++i){
    	B[i][i]=1;
	}
    while(n){
        if(n&1){
        	B=mul(B,A);
		}
        n>>=1; A=mul(A,A);
    }
    return B;
}
int n,m;
long long dp[1<<5][1<<5];
void dfs(int c,int pre,int now){
    if(c>n){
    	return;
	}
    if(c==n){
        dp[pre][now]++;
        return;
    }
    dfs(c+1,pre<<1,now<<1|1);
    dfs(c+1,pre<<1|1,now<<1);
    dfs(c+2,pre<<2,now<<2);
}
int main(){
    cin>>m>>n;
    mat a(1<<n,vec(1<<n));
    dfs(0,0,0);
    for(int i=0;i<(1<<n);i++){
        for(int j=0;j<(1<<n);j++){
        	a[i][j]=dp[i][j];
        }
    }
    a=pow(a,m+1);
    printf("%lld\n",a[0][(1<<n)-1]);
    return 0;
}
/*

in:
2 3

out:
3

*/