[hiho一下 第129周] 后缀自动机

题目来源:https://hihocoder.com/problemset/problem/1449

给定字符串s,求长度为k(需要将所有的k求出来)的子串中出现最多子串的出现次数


对于每一个节点u,当节点u为节点v的后缀链接,即u=link[v],那么以v为结尾的后缀包含在了u中。


定义F[i]为节点i后缀集合里面所有串在s里面出现的次数,显然每一个节点里所有串出现的次数都是相等的。

对于s的所有子串s1为s的第1到t (1<=t<=n) 位,给定初始值1。

拓扑排序,主动递推转移方程:F[ link[u] ] += F[u]


对于每一个节点记录该节点下最长的答案(即长度为len[]的答案),最后对每一个ans[i] = max{ans[j]} (j>=i)

#include 
#include 
#include 
#include 

#define N 2000050

using namespace std;
typedef long long LL;

char s[N];
int ch[N][27],len[N],link[N],rd[N],last,cnt,rt,n;
LL F[N],ans[N];

inline void ut(LL &x,LL y) { x = max(x,y); }

void add(int pos) {
	int x = s[pos] - 'a' + 1 , p = last , np = ++cnt;
	last = np;
	len[np] = len[p] + 1;
	F[np] = 1LL;
	
	while (p && !ch[p][x]) ch[p][x] = np , p = link[p];
	if (!p) 
		link[np] = rt;		
	else {
		int q = ch[p][x];
		if (len[q] == len[p] + 1) 
			link[np] = q;
		else {
			int nq = ++cnt;
			len[nq] = len[p] + 1;
			memcpy(ch[nq],ch[q],sizeof(ch[q]));
			link[nq] = link[q];
			link[q] = link[np] = nq;
			while (p && ch[p][x] == q) ch[p][x] = nq , p = link[p];
		}
	}
	return ;
}

int main() {
	rt = last = ++cnt;
	scanf("%s",s+1); n = strlen(s+1);
	for (int i=1;i<=n;i++) add(i);
	
	for (int i=1;i<=cnt;i++) rd[ link[i] ]++;
	
	queue q;
	for (int i=1;i<=cnt;i++) if (!rd[i]) q.push(i);
	while (!q.empty()) {
		int u = q.front(); q.pop();
		F[ link[u] ] += F[u];
		if (--rd[ link[u] ] == 0) q.push(link[u]);
		ut( ans[ len[u] ] , F[u]);
	}
	
	for (int i=n-1;i>=1;i--) ut(ans[i],ans[i+1]);
	for (int i=1;i<=n;i++) printf("%lld\n",ans[i]);
	return 0;
}


你可能感兴趣的:(后缀自动机)