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<bits/stdc++.h>
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<k;j++)
ans=(ans+dp[n][i]*dp[n][j]%Mod)%Mod;
ans=(ans*2)%Mod;
printf("%lld",ans);
return 0;
}