Bicolorings CodeForces - 1051D 计数DP

传送门

题意

给你一个2 * n的方格，每个方格可以涂成黑色或者白色，要求形成k个连通块，请问有多少种涂色方式

分析

这道题跟以前在cf做过的一个题目类似，都是去暴力枚举每一列的状态

因为只有两行，所以我们可以枚举出每一个状态:黑黑，黑白，白黑，白白，所以我们可以分成16种情况:

前面一列  后面一列 连通块数量变化
黑黑     黑黑     不变
黑白     黑黑     不变
白黑     黑黑     不变
白白     黑黑     + 1

黑黑     黑白     + 1
黑白     黑白     不变
白黑     黑白     + 2
白白     黑白     + 1

黑黑     白黑     + 1
黑白     白黑     + 2
白黑     白黑     +不变
白白     白黑     + 1

黑黑     白白     + 1
黑白     白白     +不变
白黑     白白     +不变
白白     白白     +不变

有了这些状态，很容易写出状态转移方程，代码如下

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <cstring>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define _CRT_SECURE_NO_WARNINGS
#pragma GCC optimize("Ofast","unroll-loops","omit-frame-pointer","inline")
#pragma GCC option("arch=native","tune=native","no-zero-upper")
#pragma GCC target("avx2")
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
const int INF = 0x3f3f3f3f;
const int N = 1e3 + 10;
const ll mod = 998244353;
ll dp[N][N * 2][5]; // BB  WB BW WW

int n,m;

int main(){
    scanf("%d%d",&n,&m);
    dp[1][1][1] = 1;
    dp[1][2][2] = 1;
    dp[1][2][3] = 1;
    dp[1][1][4] = 1;
    for(int i = 2;i <= n;i++){
        for(int j = 1;j <= i * 2;j++){
            dp[i][j][1] += dp[i - 1][j][1];
            dp[i][j][1] += dp[i - 1][j][2];
            dp[i][j][1] += dp[i - 1][j][3];
            if(j > 1) dp[i][j][1] += dp[i - 1][j - 1][4];
            
            if(j > 1) dp[i][j][2] += dp[i - 1][j - 1][1];
            dp[i][j][2] += dp[i - 1][j][2];
            if(j > 2) dp[i][j][2] += dp[i - 1][j - 2][3];
            if(j > 1) dp[i][j][2] += dp[i - 1][j - 1][4];
            if(j > 1) dp[i][j][3] += dp[i - 1][j - 1][1];
            if(j > 2) dp[i][j][3] += dp[i - 1][j - 2][2];
            dp[i][j][3] += dp[i - 1][j][3];
            if(j > 1) dp[i][j][3] += dp[i - 1][j - 1][4];
            if(j > 1) dp[i][j][4] += dp[i - 1][j - 1][1];
            dp[i][j][4] += dp[i - 1][j][2];
            dp[i][j][4] += dp[i - 1][j][3];
            dp[i][j][4] += dp[i - 1][j][4];
            for(int k = 1;k <= 4;k++) dp[i][j][k] %= mod;
        }
    }
    ll ans = 0;
    for(int i = 1;i <= 4;i++) ans = (ans + dp[n][m][i]) % mod;
    printf("%lld\n",ans);
    return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/