【CF1027E】Inverse Coloring（DP）

题意：给出一个n*n的矩阵，要求在每个位置涂上黑/白色，

要求满足：任意相邻的两行，其颜色要么完全相同，要么完全相反

任意相邻的两列，其颜色也要么相同要么完全相反

且这个矩形中，不存在任意一个大小大于等于k的同色矩形

求方案数模998244353

n<=5e2，1<=k<=n^2

思路：From https://blog.csdn.net/qq_34454069/article/details/81835687

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<bitset>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second 
#define MP make_pair
#define N      510
#define M      51
#define MOD 998244353
#define eps 1e-8 
#define pi     acos(-1)
#define oo     1e9

int dp[N][N],f[N];

void add(int &x,int y)
{
    x=(ll)x+y;
    if(x>=MOD) x-=MOD;
}

int main()
{
    int n,K;
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++)
    {
        dp[0][i]=1;
        for(int j=1;j<=n;j++)
         for(int k=1;k<=min(i,j);k++) add(dp[j][i],dp[j-k][i]);
    }
    for(int i=1;i<=n;i++) f[i]=(dp[n][i]-dp[n][i-1]+MOD)%MOD; 
    int ans=0;
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      if(i*j<K) add(ans,(ll)f[i]*f[j]%MOD); 
    ans=ans*2%MOD;
    printf("%d
",ans);
    return 0;
}