[JZOJ3337] 【NOI2013模拟】wyl8899的TLE

题目

题目大意

给你两个字符串$A$和$B$，可以修改$A$中的一个字符，求修改后最长的$A$的前缀，使它是$B$的子串。

---

思考历程

看到这道题之后，第一眼想到的就是后缀自动机！
$B$的子串，意味着可以把$B$建立一个后缀自动机，然后在上面跑……
刚开始的想法是将$A$在上面跑，并且试着修改后面一个字符，看看剩下的可以跑多长……
于是问题就转化成求这样的一个东西：问$A$的某个后缀从后缀自动机上的某个节点开始最多能跑多远。
我试着将这个东西预处理出来……可是没有成功……
交了个错误的程序上去，居然还有点分。

---

水法

这题不得不开“水法”这一栏目。
题目的数据真是太水了，各种暴力都能过。
最简单的暴力：枚举子串的左端点，然后暴力匹配，遇见的第一个不同就用掉修改机会，求出匹配的最大长度。
还有一种$O(n)$的正确性显然错误的DP……
跑得都比正解快……

正解

有个exkmp的方法，虽然说时间可能会有点玄学，但既然是郑姐讲的方法，就把它归为“正解”这一栏吧……
求出$A$的$next$，以$A$为模式串求出$B$的$extend$。
枚举子串的开头$i$，则后面相同的子串长度为$exten{d}_i$。
后面那个位置用修改操作，所以要从$A_{exten{d}_i+2}$和$B_{i+exten{d}_i+1}$开始继续匹配。
接下来取$\min (nex{t}_{exten{d}_i+2},exten{d}_{i+exten{d}_i+1})$。后面这个长度的串一定是相同的，因为前者等于$A$前面一段，后者也等于$A$前面一段，其中的最小值就是它们相等的长度。
然而这可能不是极限，所以后面的，就暴力判断吧……
时间应该比较优秀，虽然说那一部分暴力判断有点玄学……

然后是题解中的$LCP$做法。
先枚举子串的左端点$i$，求$LCP(B_{i..m},A)$，设为$x$。$x+1$位可以用修改机会，然后求$LCP(A_{x+2..n},B_{i+x+1..m})$，就是后面能继续延伸的长度。
当然，记得判断一下到达边界的情况。
至于求$LCP$，最典型的做法是用后缀数组，将$A$和$B$并在一起，中间用特殊字符隔开，然后求后缀数组和$height$数组，然后用数据结构维护区间$height$最小值。
还有一种做法是用后缀自动机。考虑利用$fail$指针，如果跳两个节点在$fail$树上的$LCA$，它的$len$就是它们的最长公共后缀。要求最长公共前缀，就把字符串倒过来建后缀自动机就行了……（$fail$树实际上是逆序字符串的后缀树）这是两个字符串，所以要建广义后缀自动机。

最后是许多人使用的二分和$hash$做法。
首先二分答案$k$，将$B$串中长度为$k$的子串存入$hash$表。
然后$26n$来枚举修改的方案，每次修改后再$hash$表中查询。

---

代码

后缀自动机的做法……

using namespace std;
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cassert>
#define N 50010
int n,m;
char a[N],b[N];
struct Node{
	Node *c[26],*fail;
	int len;
} d[N*4];
int cnt;
Node *S,*last;
inline void insert(int ch){
	if (last->c[ch] && last->c[ch]->len==last->len+1){
		last=last->c[ch];
		return;
	}
	Node *nw=&d[++cnt],*p;
	nw->len=last->len+1;
	for (p=last;p && !p->c[ch];p=p->fail)
		p->c[ch]=nw;
	if (!p)
		nw->fail=S;
	else{
		Node *q=p->c[ch];
		if (p->len+1==q->len)
			nw->fail=q;
		else{
			Node *clone=&d[++cnt];
			memcpy(clone,q,sizeof(Node));
			clone->len=p->len+1;
			for (;p && p->c[ch]==q;p=p->fail)
				p->c[ch]=clone;
			nw->fail=q->fail=clone;
		}
	}
	last=nw;
}
int sta[N],stb[N];
int fa[N*4][17],dep[N*4];
int get_dep(int x){
	if (dep[x])
		return dep[x];
	if (!fa[x][0])
		return dep[x]=1;
	return dep[x]=get_dep(fa[x][0])+1;
}
int LCP(int u,int v){
	if (dep[u]<dep[v])
		swap(u,v);
	for (int k=dep[u]-dep[v],i=0;k;k>>=1,++i)
		if (k&1)
			u=fa[u][i];
	if (u==v)
		return d[u].len;
	for (int i=16;i>=0;--i)
		if (fa[u][i]!=fa[v][i])
			u=fa[u][i],v=fa[v][i];
	return d[fa[u][0]].len;
}
int main(){
	scanf("%s%s",a+1,b+1);
	n=strlen(a+1),m=strlen(b+1);
	n=min(n,m);
	last=S=&d[++cnt];
	for (int i=m;i>=1;--i){
		insert(b[i]-'a');
		stb[i]=last-d;
	}
	last=S;
	for (int i=n;i>=1;--i){
		insert(a[i]-'a');
		sta[i]=last-d;
	}
	fa[1][0]=0;
	for (int i=2;i<=cnt;++i)
		fa[i][0]=d[i].fail-d;
	for (int i=1;i<=cnt;++i)
		get_dep(i);
	for (int i=1;i<=16;++i)
		for (int j=1;j<=cnt;++j)
			fa[j][i]=fa[fa[j][i-1]][i-1];
	int ans=0;
	for (int i=1;i<=m;++i){
		int lcp=LCP(stb[i],sta[1]);
		if (lcp==n || lcp==n-1){
			printf("%d\n",n);
			return 0;
		}
		ans=max(ans,lcp+(i+lcp<=m?1:0)+(i+lcp+1<=m?LCP(stb[i+lcp+1],sta[1+lcp+1]):0));
	}
	printf("%d\n",ans);
	return 0;
}

---

总结

这题应该算是一道套路题吧……
所以看题的时候，不要试着死磕一个地方，要多方面地去考虑……
比如，枚举左端点右端点什么的……