CF1178F2 Long Colorful Strip

XLVI.CF1178F2 Long Colorful Strip

首先，每一次染色，最多把一整段连续的同色格子，分成了三段。

并且，明显我们可以把连续的同色格子，直接看作一个。

这就意味着，在这么压缩后，有$m<2n$。

这就意味着$O(m^3)$的复杂度是可以接受的。

还是考虑和前一道题一样的DP。

但是这题，并非所有的$f[i,j]$都是合法的；只有对于每一种颜色，它所有的格子要么全都在段内，要么全都在段外，这样的$f[i,j]$才是合法的。因为，两个格子只要从什么时候开始颜色不一样了，那它们的颜色也会一直不一样下去。

考虑如何转移。

因为每种颜色都可能出现了不止一次，所以对于一种颜色$c$，我们有必要记录它出现的最左端$mn_c$与最右端$mx_c$。

则转移时的左右两端仍然可以采取和上一问一模一样的转移方式，即

$f[i,j]=(\sum\limits_{k=i}^{mn_{mp}}f[i,k-1]*f[k,mn_{mp}-1])*(\sum\limits_{l=mx_{mp}}^jf[mx_{mp}+1,l]*f[l+1,j])$

同时，对于区间$[mn_{mp},mx_{mp}]$内的非$mp$的所有连续格子段$[p_x,q_x]$，我们也都应该计算它们的贡献。

因此我们最终得到的是

$f[i,j]=(\sum\limits_{k=i}^{mn_{mp}}f[i,k-1]*f[k,mn_{mp}-1])*(\sum\limits_{l=mx_{mp}}^jf[mx_{mp}+1,l]*f[l+1,j])*\prod f[p_x,q_x]$

复杂度仍是$O(n^3)$。

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
int n,m,num[1000100],f[1010][1010],mn[1010],mx[1010];
int main(){
	scanf("%d%d",&n,&m);
	memset(mn,0x3f3f3f3f,sizeof(mn));
	for(int i=1;i<=m;i++){
		scanf("%d",&num[i]);
		if(num[i]==num[i-1])i--,m--;
	}
	if(m>2*n){puts("0");return 0;}
//	for(int i=1;i<=m;i++)printf("%d ",num[i]);puts("");
	for(int i=1;i<=m;i++)mx[num[i]]=max(mx[num[i]],i),mn[num[i]]=min(mn[num[i]],i);
//	for(int i=1;i<=n;i++)printf("%d %d\n",mx[i],mn[i]);
	for(int i=1;i<=m+1;i++)for(int j=0;j<i;j++)f[i][j]=1;
	for(int l=1;l<=m;l++)for(int i=1,j=i+l-1;j<=m;i++,j++){
		int mp=0x3f3f3f3f;
		for(int k=i;k<=j;k++)mp=min(mp,num[k]);
		if(mn[mp]<i||mx[mp]>j)continue;
		int A=0,B=0;
		for(int k=mn[mp];k>=i;k--)(A+=(1ll*f[i][k-1]*f[k][mn[mp]-1]%mod))%=mod;
		for(int l=mx[mp];l<=j;l++)(B+=(1ll*f[mx[mp]+1][l]*f[l+1][j]%mod))%=mod;
		f[i][j]=1ll*A*B%mod;
//		printf("(%d,%d):\n",i,j);
		for(int p=mn[mp]+1,q=mn[mp];p<mx[mp];){
			while(q<j&&num[q+1]!=mp)q++;
//			printf("(%d,%d)\n",p,q);
			f[i][j]=1ll*f[i][j]*f[p][q]%mod;
			q++,p=q+1;
		}
//		printf("%d\n",f[i][j]);
	}
	printf("%d\n",f[1][m]);
	return 0;
}