【SPOJ8222】Substrings【后缀自动机】

万能VJudge

后缀自动机总结

求每种长度的子串最大数量。

fail树上加起来即可。

注意：初始size只能打在字符结点上，不能打在复制结点上。这违背了size的本意。

#include<bits/stdc++.h>
using namespace std;

#define int long long

char ch[250003];
struct node{
	int ch[26],len,fail,size;
}t[500003];
int last=1,cnt=1;
void insert(int x){
	int p=last;int now=++cnt;last=cnt;t[now].len=t[p].len+1;t[now].size=1;
	for(;p&&!t[p].ch[x];p=t[p].fail)t[p].ch[x]=now;
	if(!p)t[now].fail=1;
	else{
		int q=t[p].ch[x];if(t[q].len==t[p].len+1)t[now].fail=q;
		else{
			int tem=++cnt;t[tem]=t[q];t[q].fail=t[now].fail=tem;t[tem].len=t[p].len+1;t[tem].size=0;
			for(;p&&t[p].ch[x]==q;p=t[p].fail)t[p].ch[x]=tem;
		}
	}
}
int ans[250003];
int first[500003],nxt[500003],to[500005],tot;
void add(int a,int b){
	nxt[++tot]=first[a];first[a]=tot;to[tot]=b;
}
void dfs(int u){
	for(int i=first[u];i;i=nxt[i]){
		int v=to[i];dfs(v);
		t[u].size+=t[v].size;
	}ans[t[u].len]=max(ans[t[u].len],t[u].size);
}
signed main(){
	scanf("%s",ch+1);
	int len=strlen(ch+1);
	for(int i=1;i<=len;i++)insert(ch[i]-'a');
	for(int i=2;i<=cnt;i++)add(t[i].fail,i);
	dfs(1);
	for(int i=1;i<=len;i++)cout<<ans[i]<<'\n';
	return 0;
}