CF1051D Bicolorings

题目描述

咳咳，懒得复制了上面是两张图：）

解题思路

这题是一道很好的题，感觉之前做过，一开始手推状态找规律，可以用状压但是没想到

借鉴了一下大佬的dp

modify数组用以累加新增的状态数

dp数组表示前i列第j个连通块，a，b表示该列的状态，转移方程见代码

下面就是美丽的代码了

AC Code

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define P (LL)(998244353)

typedef long long LL;
typedef long double ld;
typedef unsigned long long ULL;

using namespace std;

const int MAXN =  + ;
const int MAXK = MAXN << ;
int dp[MAXN][MAXK][][];
int modify[][][][];
void pre() {
    modify[][][][] = ; modify[][][][] = ; modify[][][][] = ; modify[][][][] = ;
    modify[][][][] = ; modify[][][][] = ; modify[][][][] = ; modify[][][][] = ;
    modify[][][][] = ; modify[][][][] = ; modify[][][][] = ; modify[][][][] = ;
    modify[][][][] = ; modify[][][][] = ; modify[][][][] = ; modify[][][][] = ;
}
int main() {
    int n , k;
    cin>>n>>k;
    pre();
    dp[][][][] = dp[][][][] = ;
    dp[][][][] = dp[][][][] = ;

    for(int i = ; i <= n; i++)
        for(int j = ; j <= k; j++)
            for(int a = ; a < ; a++)
                for(int b = ; b < ; b++)
                    for(int A = ; A < ; A++)
                        for(int B = ; B < ; B++) {
                            int block = modify[A][B][a][b];
                            if(j >= block) dp[i][j][a][b] = (dp[i][j][a][b] + dp[i - ][j - block][A][B]) % P;
                        }
    int ans = ;
        for(int A = ; A < ; A++)
            for(int B = ; B < ; B++)
                ans = (ans + dp[n][k][A][B]) % P;
    cout<<ans;
    return ;
}