[CF1178 F2] Long Colorful Strip

F2 - Long Colorful Strip

很牛的题!

首先,我们可以将颜色相同的一段区间缩成一个点,那么每次加入一个新的颜色时,最多只能将其所覆盖的那个颜色所属的区间分成三部分(原本:00000000,加入1后\(\rightarrow\)0001111000),也就是增加了两个点,那么也就意味着最终所成的点的个数最多只有\(2*n-1\)个,这样的话,\(m\)看着很大,其实也就是\(O(N)\)级别的

对于这种区间覆盖的题,优先考虑区间\(dp\)

设\(dplr\)表示让区间\(l,r\)合法的方案数,且满足\(l,r\)中的颜色不会在\(l,r\)外出现

设\(p\)为区间\(l,r\)中最小的颜色,其出现的最早的位置为\(mn_p\),最晚的位置为\(mx_p\),那么显然有我们可以一开始就让\(p\)覆盖区间\(a,b\)(\(mn_p,mx_p\ina,b\)),那么就有转移式

\dp\[lr=\sum_{a\inl,mn_p}\sum_{b\inmx_p,r}dpla-1\times dpab\times dpb+1r \]

其中,对于\(dpab\),其实就是相当于对于\(a,b\)的所有方案,直接在每种方案的第一个操作之前给\(a,b\)覆盖上颜色\(p\)即可(总之就是让第一步变成用\(p\)覆盖\(a,b\),具体的,若原本的方案没有这一步,加上即可;否则,若其原本只用\(p\)覆盖\(s,t\)(\(s,t\ina,b\),因为\(p\)也是\(a,b\)中最小的值,所以原本的用\(p\)覆盖\(s,t\)的这一步也一定是该方案中的第一步)将其补全成覆盖\(a,b\)即可

但是可以发现,当\(a=l,b=r\)时,就转不了了,所以考虑将\(dpab\)拆开

因为\(p\)的最早出现位置为\(mn_p\),最晚出现位置为\(mx_p\),所以显然有所有处于\(mn_p,mx_p\)中的颜色的出现区间范围也不会超过\(mn_p,mx_p\),这也意味这\(mn_p,mx_p\)将\(a,b\)分隔成了三个区间\(a,mn_p-1,mn_p,mx_p,mx_p+1,b\),这三个区间是"独立的",即它们的颜色集合没有交集

证明的话可以从\(mn_p\)和\(mx_p\)这两个点下手,因为它们的颜色\(p\)是区间\(a,b\)中最小的颜色,这就意味着区间\(a,b\)中剩下的所有颜色都不会跨越它们,也就是没有颜色会覆盖上述三个区间中的任意两个,所以所有颜色只会待在对应的一个区间内

\dp\[ab=dpamn_p-1\times dpmx_p+1b\times \sum_{i,j}dpij \]

其中\(i,j\)表示所有不包含颜色\(p\)的、属于区间\(mn_p,mx_p\)的极长区间,它实际上如何转移的和上文中\(对dpab\)的解释一致

所以我们就得到了最终的式子

\dp\[lr=\sum_{a\inl,mn_p}\sum_{b\inmx_p,r}dpla-1\times dpamn_p-1\times(\sum_{i,j}dpij)\times dpmx_p+1b\times dpb+1r \]

复杂度\(O(N^3)\)

代码中,只要保证了当前区间\(l,r\)的最小颜色\(p\)的范围属于当前区间,在转移时因为其他转移给\(dplr\)的\(dpxy\)会保证属于\(x,y\)的所有颜色只属于\(x,y\)中,所以最终就一定能保证\(l,r\)中的所有颜色只属于\(l,r\)中

也就是归纳证明即可

c++ 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=505,M=1e6+5,MOD=998244353;
int n,m,col[M],dp[N<<1][N<<1],mn[N],mx[N];
void add(int &x,int y){ x+=y; if(x>=MOD) x-=MOD; }
int ad(int x,int y){ x+=y; if(x>=MOD) x-=MOD; return x; }
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;++i){
		scanf("%d",&col[i]);
		if(col[i]==col[i-1]) --i,--m;
		else (!mn[col[i]])&&(mn[col[i]]=i),mx[col[i]]=i;
	}
	if(m>=(n<<1)) return puts("0"),0;
	for(int i=1;i<=m+1;++i) for(int j=0;j<i;++j) dp[i][j]=1;
	for(int len=1;len<=m;++len)
		for(int l=1,r,p,a,b;l+len-1<=m;++l){
			r=l+len-1,p=N,a=0,b=0;
			for(int k=l;k<=r;++k) p=min(p,col[k]);
			if(mn[p]<l||mx[p]>r) continue;
			for(int k=l-1;k<=mn[p]-1;++k) add(a,1ll*dp[l][k]*dp[k+1][mn[p]-1]%MOD);
			for(int k=mx[p];k<=r;++k) add(b,1ll*dp[mx[p]+1][k]*dp[k+1][r]%MOD);
			dp[l][r]=1ll*a*b%MOD;
			for(int i=mn[p]+1,j=mn[p];i<mx[p];i=(j+=2)){
				for(;col[j+1]!=p;++j) ;
				dp[l][r]=1ll*dp[l][r]*dp[i][j]%MOD;
			}
		}
	printf("%d",dp[1][m]);
	
	return 0;
}