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<k然后转移下一行染成一样的还是相反的就好了，然而GB爆炸1小时50多分用n^2的复杂度A掉，一发上橙，巨强无比。感觉打acm之后DP做得太少了，好多方法都想不起来了菜哭.jpg

#include<cstdio>
#include<cstring>
#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<k;len++)
	{
		for(int i=1;i<=n;i++)
			for(int j=1;j<=i;j++)
				g[i][j]=0;
		g[1][1]=sum[len];
		for(int i=1;i<=n-1;i++)
			for(int j=1;j<=i && j*len<k;j++)
			{
				if((j+1)*len<k)
					add(g[i+1][j+1],g[i][j]);
				add(g[i+1][1],g[i][j]);
			}
		for(int j=1;j<=n && j*len<k;j++)
			add(ans,g[n][j]);
	}
}

inline void print()
{
	printf("%d",ans);	
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}