Codeforces R1027E Inverse Coloring 【DP】

Codeforces R1027E Inverse Coloring

---

题目链接

---

题目大意是给你一个n*n的棋盘，然后你需要将它黑白染色
相邻的行列要么完全相同，要么完全不相同
然后不能出现同色矩形包含大于等于k个色块

好性质:
所以我们只需要一行一列的状态就可以确定所有格子的状态

然后我们考虑DP啊， dp[i][j] d p [ i ] [ j ] 表示长度为i的一行、列最长连续同色不超过j的方案数

这个DP就很显然了，枚举一下当前色段的长度就好了

然后最后我们需要差分回去，这个细节可以好好想想
然后可以得到 dp[n][j] d p [ n ] [ j ] 表示长度是n且恰好最长色段为j的方案数

然后枚举统计答案，最后乘上2

---

#include
using namespace std;
#define N 1010
#define LL long long
#define Mod 998244353
int n,k;
LL dp[N][N],ans=0;
LL sum[N][N];
int main(){
    cin>>n>>k;
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++)
            for(int p=1;p<=j;p++)
                dp[i][j]=(dp[i][j]+dp[i-p][min(j,i-p)])%Mod;
    for(int i=n;i>=1;i--)dp[n][i]=(dp[n][i]-dp[n][i-1]+Mod)%Mod;
    for(int i=1;i<=n;i++)
        for(int j=1;j*i*dp[n][j]%Mod)%Mod;
    ans=(ans*2)%Mod;
    printf("%lld",ans);
    return 0;
}