CF1393E2 - Twilight and Ancient Scroll (harder version)

题目大意

给定$n$个串$S_i$，求在每个串中至多删除一个字符（可以删到空）

最终$S’_i$字典序单调不减的方案数

dp

显然是记录上一个串删除的位置$j$，得到$dp_j$，依次考虑每个串

设$S$为当前串，$T$为上一个串

每次我们枚举当前串删除的位置$i$，不妨设得到的新串为$S’$，考虑对于$S’$找到所有合法转移

设$L=LCP(S,T)$，在$T$中删除位置为$t$

1.$t>L+1$

则可以根据$T_{L+1}$和$S’_{L+1}$的关系确定$T’$与$S’$的关系，只需要处理一个后缀和$suf_i$

2.设$p$为$T_{L+1}$向前延伸且字符相同的最长位置，$t\in[p,L+1]$

此时，删除$T_t$可能会使得$L’>L$，所以提出来特殊处理

显然$t\in[p,L+1]\Leftrightarrow t=L+1$，那么对于得到的新串可以再求$LCP(S’,T’)$来判断大小

只需要一个区间和

3.$t<p$，此时$L’=LCP(S’,T’)<L$

由于$L’<L$,比较$S’,T’$相当于比较$T_{t:},T_{t+1:}$

其中$T_{t:}$表示$T$在$t$开始的后缀，那么预处理找到所有合法的$t$，累前缀和即可得到$pre_{p-1}$

关于实现

写$\text{SA,SAM}$就完蛋了

由于删除一个字符之后，求$LCP$的两个串就是在原串基础上进行$\pm 1$的偏移

线性预处理三个$LCP$数组即可，当然真正求的时候需要分类讨论一下(真的，就一下/md)

判断$T_{t:},T_{t+1:}$容易倒着线性预处理出来，在代码里是$chk_i$

所有操作均可以线性处理，时间复杂度为$O(L)$，77ms

Tips:

比较过程中容易出现奇妙的越界，我写得很丑

const int N=1e6+10,P=1e9+7;

int n,m;
char A[2][N],*S=A[0],*T=A[1];
int X[N],Y[N],Z[N];
int dp[N],F[N],suf[N],pre[N],chk[N],L[N];

int main(){
	rep(_,1,rd()) {
		swap(S,T),swap(n,m);
		scanf("%s",S+1),n=strlen(S+1);
		rep(i,n+1,m+2) S[i]=1;
		rep(i,m+1,n+2) T[i]=-1;
		if(!m) {
			rep(i,1,n+1) dp[i]=1;
			continue;
		}
		rep(i,max(n,m)+1,i+2) X[i]=Y[i]=Z[i]=0;
		drep(i,max(n,m),1) {
			X[i]=S[i]==T[i]?X[i+1]+1:0;
			Y[i]=S[i]==T[i+1]?Y[i+1]+1:0;
			Z[i]=S[i+1]==T[i]?Z[i+1]+1:0;
		}

		// initiate
		suf[m+2]=0;
		drep(i,m+1,1) suf[i]=suf[i+1]+dp[i],Mod1(suf[i]);
		drep(i,m,1) chk[i]=T[i]!=T[i+1]?T[i]>T[i+1]:chk[i+1];
		rep(i,1,m+1) L[i]=T[i]==T[i-1]?L[i-1]:i;
		rep(i,1,m) pre[i]=pre[i-1]+chk[i]*dp[i],Mod1(pre[i]);


		rep(i,1,n+1) {
			int l=min(X[1],i-1);
			if(l==i-1) l+=Z[i];
			F[i]=0;

			auto I=[&](int x){ return (x>=i)+x; };

			// type1 , deleted pos > l+1
			if(T[l+1]<=S[I(l+1)]) F[i]+=suf[l+2],Mod1(F[i]);
			int p=L[l+1];
			
			// type2 ,delete T between [p..l+1]  ,the same as we delete T[l+1]
			int r=0; // r= LCP(S'[l+2],T[l+2]) ,check if delete T[l+1], T'<=S'
			if(l+1<i) r+=min(i-l-1,Y[l+1]);
			if(r==max(0,i-l-1)) r+=X[l+2+r];
			if(T[l+2+r]<=S[I(l+1+r)]) F[i]=(F[i]+0ll+suf[p]-suf[l+2]+P)%P;
			
			// type3 , delete T at [1,p-1], so LCP'<l , and we determine T'<=S' by chk[i]
			F[i]=(F[i]+pre[p-1])%P;
		}
		rep(i,1,n+1) dp[i]=F[i];
	}
	int ans=0;
	rep(i,1,n+1) ans+=dp[i],Mod1(ans);
	printf("%d\n",ans);
}