spoj 8222 substrings 【后缀自动机】

来源:互联网 发布:2017软件设计师 编辑:程序博客网 时间:2024/06/06 12:57

解题思路:

建立sam,那么一个节点i所对应的字符串出现次数即为其right集合大小,而其包含字符串长度为(min[i],max[i]),且随着parent链代表字符串长度减小,其出现次数单增。所以先令f[max[i]]=right[i],最后用f[i]更新f[i-1]即可。

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>#include<cmath>#include<ctime>#include<queue>#include<vector>#include<set>#include<map>#define ll long longusing namespace std;const int N=500005;int n,tot=1,last=1;int son[N][26],r[N],mx[N],fa[N],f[N],c[N],q[N];char s[N];void extend(int c){    int p=last,np=last=++tot;    mx[np]=mx[p]+1,r[np]=1;    while(p&&!son[p][c])son[p][c]=np,p=fa[p];    if(!p)fa[np]=1;    else    {        int q=son[p][c];        if(mx[q]==mx[p]+1)fa[np]=q;        else        {            int nq=++tot;            mx[nq]=mx[p]+1;            memcpy(son[nq],son[q],sizeof(son[q]));            fa[nq]=fa[q];            fa[q]=fa[np]=nq;            while(son[p][c]==q)son[p][c]=nq,p=fa[p];        }    }}int main(){    //freopen("lx.in","r",stdin);    scanf("%s",s);n=strlen(s);    for(int i=0;i<n;i++)extend(s[i]-'a');    for(int i=1;i<=tot;i++)c[mx[i]]++;    for(int i=1;i<=n;i++)c[i]+=c[i-1];    for(int i=1;i<=tot;i++)q[c[mx[i]]--]=i;    for(int i=tot;i>=1;i--)r[fa[q[i]]]+=r[q[i]];    for(int i=1;i<=tot;i++)f[mx[i]]=max(f[mx[i]],r[i]);    for(int i=n;i>=1;i--)f[i]=max(f[i],f[i+1]);    for(int i=1;i<=n;i++)cout<<f[i]<<'\n';    return 0;}