[XSY] 相似（DP套DP）

相似

在看这道题前，有必要先看一下DP套DP的入门题[uoj3864]Hero meet devil，附上两篇写得不错的题解：
https://blog.youkuaiyun.com/Ike940067893/article/details/87863041
https://www.cnblogs.com/RabbitHu/p/BZOJ3864.html

• 可以发现，$S$和$T$相似，等价于它们的最长公共子序列长度至少为$n-k$
• 考虑如何求两个字符串的LCS。设 $lc{s}_{i,j}$ 表示 $S[1...i]$ 与 $T[1...j]$ 的 LCS 长度，转移显然为
  l c s i , j = { l c s i − 1 , j − 1 + 1 ( S [ i ] = = T [ j ] ) m a x { l c s i − 1 , j , l c s i , j − 1 } ( S [ i ] ! = T [ j ] ) lcs_{i,j}=\left\{ \begin{aligned} &lcs_{i-1,j-1}+1(S[i]==T[j]) \\ & max\{lcs_{i-1,j},lcs_{i,j-1}\} (S[i]!=T[j])\\ \end{aligned} \right. lcsi,j​={​lcsi−1,j−1​+1(S[i]==T[j])max{lcsi−1,j​,lcsi,j−1​}(S[i]!=T[j])​
• 考虑如何拓展到$T$任意的情况。自然的想法是DP套DP：设$dp[j][lc{s}_{1,j},lc{s}_{2,j},...,lc{s}_{n,j}]$表示 考虑到$T$的前$j$位，$lc{s}_{-,j}$的情况如第二维所示 的方案数
• 直接这样维护状态肯定是行不通的，考虑优化：
  1.发现 l c s i , j − l c s i − 1 , j ∈ { 0 , 1 } lcs_{i,j}-lcs_{i-1,j}\in\{0,1\} lcsi,j​−lcsi−1,j​∈{0,1}，那么 $lc{s}_{1,j},lc{s}_{2,j},...,lc{s}_{n,j}$的差分数组必是一个01 串 ，我们 状压这个01串 来描述第二维的状态
  2.$n$位01串还是太大了，考虑如何利用$k$很小这个性质进行优化。
  考 虑 某 个 l c s i , j ， 若 ∣ i − j ∣ > k ， 那 么 在 转 移 到 l c s n , n 时 ， 必 然 值 不 超 过 l c s i , j + m i n { n − i , n − j } < = m i n { i , j } + m i n { n − i , n − j } = n − ∣ i − j ∣ < n − k \color{Red}{考虑某个lcs_{i,j}}，若|i-j|>k，那么在转移到lcs_{n,n}时，必然值不超过lcs_{i,j}+min\{n-i,n-j\}<=min\{i,j\}+min\{n-i,n-j\}=n-|i-j|<n-k 考虑某个lcsi,j​，若∣i−j∣>k，那么在转移到lcsn,n​时，必然值不超过lcsi,j​+min{n−i,n−j}<=min{i,j}+min{n−i,n−j}=n−∣i−j∣<n−k
  因此这样的$lc{s}_{i,j}$是没用的，可以直接丢掉，即对于每个$j$，我们只需要记录$lc{s}_{j-k,j},lc{s}_{j-k+1,j},...,lc{s}_{j+k,j}$
  3.只知道$lc{s}_{j-k,j}$~$lc{s}_{j+k,j}$的差分数组，我们是无法知道它们的实际值的，因此考虑开多一维来记录$lc{s}_{j,j}$的实际值以推出其它的$lcs$值，即我们设$dp[j][x][s]$表示 考虑到$T$的前$j$位，$lc{s}_{j,j}==x$，$lc{s}_{j-k,j},lc{s}_{j-k+1,j},...,lc{s}_{j+k,j}$的差分数组状压后为$s$ 的方案数
  4.$\because lc{s}_{n,n}\in [n-k,n](即不匹配的字符最多k个)$
  $\therefore \forall j\in [1,n]满足lc{s}_{j,j}\in [j-k,j]$，所以我们可以将$dp$的第二维设置为$lc{s}_{j,j}-(j-k)=lc{s}_{j,j}-j+k$，即$dp[][x][]$表示$lc{s}_{j,j}-j+k==x$，这样就令$x\in [0,k]$
• 考虑如何实现状态转移。$dp$数组的第三维不好直接得到，我们考虑把它预处理出来：设$p[i][j][s1][c]$表示考虑到$T$的前$i$位，$lc{s}_{i,i}==j$（知道$lc{s}_{i,i}$就可以通过差分数组推出其它的$lcs$值），$lc{s}_{i-k,i},lc{s}_{i-k+1,i},...,lc{s}_{i+k,i}$的差分数组状压后为s1，在$T$的$i+1$位新增字符$c$后，$lc{s}_{i+1-k,i+1},lc{s}_{i+2-k,i+1},...,lc{s}_{i+1+k,i+1}$状压完的差分数组
• 同样，需要优化$p$的状态设置：
  1.同上，我们可以将$p$的第二维设置为$lc{s}_{i,i}-(i-k)=lc{s}_{i,i}-i+k$，即$p[][j][][]$表示$lc{s}_{i,i}-i+k==j$，这样就令$j\in [0,k]$
  2.其实$p$转移到的新差分数组状态与新增字符具体是什么无关，与新增字符与S每一位的匹配情况有关。再进一步地，
  $\because lc{s}_{h,h}\in [h-k,h]$
  $\therefore T[h]只可能与S[h-k]$~$S[h+k]匹配$，即我们只需要考虑新增字符与$S[i+1-k]$~$S[i+1+k]$的匹配情况。直接把第四维设置成这个匹配情况即可。
  3.仔细想想，把第四维改成匹配情况后，$p$的转移已经与第一维无关了，直接删掉即可

至此，这道题就可以实现了

#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 998244353
using namespace std;
typedef pair<int,int> pr;
const int N=30001;
const int K=5;
const int S=256;
int n,m,ans,dp[N][K][S],f[31],g[31];
pr p[K][S][S<<1];
char ch[N];
bool flag[31];
int add(int a,int b){
	a=a+b;
	if(a>mod) a-=mod;
	return a;
} 
int mul(int a,int b){
	return 1ll*a*b%mod;
}
void dfs(int a,int b,int c){
	if(dp[a][b][c]==0) return;
	int l=max(a-m,0),r=min(a+m,n-1),sz=26;
	for(int i=l;i<=r;i++){
		if(!flag[ch[i+1]-'A']){
			flag[ch[i+1]-'A']=1;
			sz--;
		}
		f[ch[i+1]-'A']|=(1<<(i-a+m));//统计每种字符的匹配情况 
	}
	for(int i=l;i<=r;i++){
        if(flag[ch[i+1]-'A']){
            int x=p[b][c][f[ch[i+1]-'A']].first;
			int y=p[b][c][f[ch[i+1]-'A']].second;
            if(x>=0) dp[a+1][x][y]=add(dp[a+1][x][y],dp[a][b][c]);
            flag[ch[i+1]-'A']=0;
            f[ch[i+1]-'A']=0;
        }
    }
    int x=p[b][c][0].first,y=p[b][c][0].second;
    if(x>=0) dp[a+1][x][y]=add(dp[a+1][x][y],mul(dp[a][b][c],sz));
}
int main(){
	scanf("%s%d",ch+1,&m);
	n=strlen(ch+1);
	//省略一维l，表示考虑到第l位 
	for(int i=0;i<=m;i++){//lcs[l][l]-l+m==i 
		for(int j=0;j<(1<<(2*m));j++){//lcs[l-m][l]~lcs[l+m][l]的差分数组状压 
			for(int k=0;k<(1<<(2*m+1));k++){//T[l+1]与S[l+1-m]~S[l+1+m]匹配情况状压 
				//f:(lcs[l-m][l]-l+m)~(lcs[l+m][l]-l+m) 
				//g:(lcs[l+1-m][l+1]-l+m)~(lcs[l+1+m][l+1]-l+m)
				f[m]=i;
				for(int a=m-1;a>=0;a--) f[a]=f[a+1]-(j>>a&1);
				for(int a=m;a<2*m;a++) f[a+1]=f[a]+(j>>a&1);
				for(int a=0;a<=2*m;a++){
                    if((k>>a)&1) g[a+1]=max(f[a]+1,f[a+1]);//匹配
                    else g[a+1]=max(f[a],f[a+1]); //不匹配 
                    g[a+1]=max(g[a],g[a+1]);
                }
                int x=g[m+1]-1;//x=lcs[l+1][l+1]-(l+1)+m
                int y=0;
                if(x>=0){
                    for(int a=m-1;a>=0;a--) 
						if(g[a+1]^g[a+2]) y|=1<<a;
                    for(int a=m;a<2*m;a++) 
						if(g[a+1]^g[a+2]) y|=1<<a;
                }
                p[i][j][k]=make_pair(x,y);
			}
		}
	}//预处理 
	for(int i=0;i<=30;i++) f[i]=0;
	dp[0][m][0]=1;
	for(int i=0;i<n;i++){//考虑到第i位 
        for(int j=0;j<=m;j++){//lcs[i][i]-i+m==j 
            for(int k=0;k<(1<<(2*m));k++){//差分数组状压 
                dfs(i,j,k);
            }
        }
    }//dp
    for(int i=0;i<=m;i++){
        for(int j=0;j<(1<<(2*m));j++){
            ans=add(ans,dp[n][i][j]);
        }
    }
    printf("%d\n",ans);
	return 0;
}