P6793 [SNOI2020]字符串（后缀树上DP）

P6793 [SNOI2020]字符串

给出两个长度为\(n\)的字符串\(a,b\)，取出他们所有长为\(k\)的子串（各有\(n-k+1\)个），这些子串分别组成集合\(A,B\)。

现在要修改\(A\)中的串，使得\(A\)和\(B\)完全相同。

可以任意次选择修改\(A\)中的一个串的一段后缀。花费为这段后缀的长度。

总花费为每次修改花费之和。

求总花费的最小值。

做法：

将\(a\)和\(b\)的后\(n-k+1\)个字符倒序插入SAM。

然后建出link树，\(a\)和\(b\)的后缀在树上对应若干个节点。

然后对于两个节点，他们的lca越深匹配效果越好。

因为匹配代价是\(k-len(lca)\)。

问题转化为：

给出树上两个相同的点集。

两个点匹配的代价是\(min(k,dep(lca))\)。

询问怎么匹配，总代价最小。

对每个子树维护两个值：

\(f[u][0]\)表示点集\(A\)剩余未匹配的点数。

\(f[u][1]\)表示点集\(B\)剩余未匹配的点数。

自底向上转移即可。

时间复杂度\(O(n)\)。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int n,k,nxt[maxn][26],link[maxn],len[maxn],dp[maxn][2],tot=1,lst=1;
string A,B;
void sam_extend (char c,int f,int v) {
	int cur=++tot,p=lst;
	dp[cur][f]=v;
	len[cur]=len[lst]+1;
	while (p&&!nxt[p][c-'a']) {
		nxt[p][c-'a']=cur;
		p=link[p];
	}
	if (!p) link[cur]=1;
	else {
		int q=nxt[p][c-'a'];
		if (len[p]+1==len[q]) link[cur]=q;
		else {
			int clone=++tot;
			len[clone]=len[p]+1;
			for (int i=0;i<26;i++) nxt[clone][i]=nxt[q][i];
			link[clone]=link[q];
			while (p&&nxt[p][c-'a']==q) {
				nxt[p][c-'a']=clone;
				p=link[p];
			}
			link[q]=link[cur]=clone;
		}
	}
	lst=cur;
}
vector<int> g[maxn];
long long ans=0;
void dfs (int u) {
	for (int v:g[u]) {
		dfs(v);
		for (int i=0;i<2;i++) dp[u][i]+=dp[v][i];
	}
	int gg=min(dp[u][0],dp[u][1]);
	ans+=1ll*gg*max(0,k-len[u]);
	dp[u][0]-=gg;
	dp[u][1]-=gg;
}
int main () {
	ios::sync_with_stdio(false);
	cin>>n>>k;
	cin>>A;
	cin>>B;
	for (int i=n-1;i>=0;i--) {
		int ff=0;
		if (i<=n-k) ff=1;
		sam_extend(A[i],0,ff);
	}
	lst=1;
	for (int i=n-1;i>=0;i--) {
		int ff=0;
		if (i<=n-k) ff=1;
		sam_extend(B[i],1,ff); 
	}
	for (int i=2;i<=tot;i++) {
		g[link[i]].push_back(i);
	}
	dfs(1);
	cout<<ans;
}