牛客小白月赛7 J 方格填色(状压DP+矩阵快速幂)

题目链接：https://www.nowcoder.com/acm/contest/190/J

 

题目大意：给一个m*n的方格，每个格子可以是黑色或白色，要求左右相邻两格不能同时为白色，且相邻两列不能全为黑色，问可以有几种情况数。

 

题目思路：由于数据范围1<=m<=5,1<=n<=1e18，所以很容易看出是状压dp。由于他要求的都是对于两列之间，也就是说行之间没有要求，所以我们只要在列之间进行转移即可。这里使用状态来表示每列的情况。用0表示黑色用1表示白色。那么由于左右相邻两格不能同时为白色，所以表示两种状态的十进制数进行与得到的结果需要为0，还有一个条件是相邻两列不能全为黑色，那么相邻两列中至少有一个状态非0。然后我们构造一个2^m*2^m的矩阵，行和列分别表示每种情况，然后数字中只有1和0，表示这两种状态可不可以进行转移。然后可以转移的乘以a[i-1][0],a[i-1][1]……a[i-1][2^m-1](竖着的)，就能获得a[i][0],a[i][1]……a[i][2^m-1](竖着的)表示第i列各种状态的情况个数。然后全部加起来就是答案了。

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long 
const int MAXN = 1e5+5;
const int MOD = 1e9+7;
int t[32][32];
struct node
{
    int n,m,a[32][32];
    void operator*=(const node& y)
    {
        memset(t,0,sizeof(t));
        int i,j,k;
        for(i=0;i<n;i++)for(k=0;k<m;k++)for(j=0;j<y.m;j++)t[i][j]=(t[i][j]+(ll)a[i][k]*y.a[k][j])%MOD;
        m=y.m;
        for(i=0;i<n;i++)for(j=0;j<m;j++)a[i][j]=t[i][j];
    }
}A,S,T;
int main(){
    int m;
    ll n;
    while(~scanf("%d%lld",&m,&n)){
        memset(A.a,0,sizeof(A.a));
        memset(S.a,0,sizeof(S.a));
        memset(T.a,0,sizeof(T.a));
        m=1<<m;
        A.n=A.m=S.n=S.m=T.n=m;
        T.m=1;
        rep(i,0,m-1){
            A.a[i][i]=1;
            T.a[i][0]=1;
        }
        rep(i,0,m-1){
            rep(j,0,m-1){
                if(!(i&j)&&(i|j)){
                    S.a[i][j]=1;
                }
            }
        }
        for(n--;n;n>>=1,S*=S)if(n&1)A*=S;
        A*=T;
        ll ans=0;
        rep(i,0,m-1){
            ans=(ans+A.a[i][0])%MOD;
        }
        printf("%lld\n",ans);
    }
    return 0;
}