codeforces 1027E Inverse Coloring

比赛的时候一直想不到怎么预处理出第一排的情况，只要知道第一排最长长度为len(len=1...n)的情况有多少种就行了，一直想着容斥之类的，后来rols说艹被卡空间了，我突然发现好像n^3DP好像就行。。。。然后滚动数组优化掉一维，f[i][j][k]表示第一行到第i列，最长长度为j，当前相同颜色到末尾的长度为k的方案数，那么第i+1列添加同样颜色或不同颜色，进行转移。i+1只由i转移过来，所以可以优化掉第1维，那么就可以求出sum[j]表示第一行最长长度为j的方案数有多少，枚举每一个len=1...n，g[i][j]表示第1行放一个最长长度为len的染色方案，第i行，已有j行连续的方案数，每次染下一行只要判断一蛤(j+1)*len

#include
#include
#define maxl 510
#define mod 998244353

int n,k,ans;
int f[2][maxl][maxl],sum[maxl];
int g[maxl][maxl];

inline void add(int &x,int y)
{
	x=(x+y)%mod;
}

inline void prework()
{
	scanf("%d%d",&n,&k);
	f[0][0][0]=1;f[1][1][1]=2;
	for(int i=1;i<=n-1;i++)
	{
		for(int j=1;j<=i+1;j++)
			for(int k=1;k<=j;k++)
				f[(i+1)&1][j][k]=0;
		for(int j=1;j<=i;j++)
			for(int k=1;k<=j;k++)
			{
				if(k==j)
					add(f[(i+1)&1][j+1][k+1],f[i&1][j][k]);
				else
					add(f[(i+1)&1][j][k+1],f[i&1][j][k]);
				add(f[(i+1)&1][j][1],f[i&1][j][k]);
			}
	}
	for(int j=1;j<=n;j++)
		for(int k=1;k<=j;k++)
			add(sum[j],f[n&1][j][k]);
}

inline void mainwork()
{
	ans=0;
	for(int len=1;len<=n && len