后缀自动机 重复旋律 5(求所有本质不同的子串)

来源:互联网 发布:spring mvc传对象数组 编辑:程序博客网 时间:2024/05/22 03:21

后缀自动机二 重复旋律五

本题要求我们求出一个字符串S中本质不同的子串个数。

显然,答案就是所有状态上的子串个数之和,这里说的一个状态上的子串个数,是longest-shortest+1。主要需要学习的是,后缀自动机的O(N)构建方法。需要了解的是,一个状态u的shortest=fail.longest+1。

#include <bits/stdc++.h>using namespace std;const int mxn = 2000100;namespace suffixAuttomaton{    int tail=2;    int fail[mxn];    int maxi[mxn];    int mini[mxn];    int next[mxn][26];    inline int extend(int p,int c)    {        int t=tail++;        maxi[t]=maxi[p]+1;        while(p&&!next[p][c])            next[p][c]=t,p=fail[p];        if(p)        {            int q=next[p][c];            if(maxi[q]==maxi[p]+1)                fail[t]=q,mini[t]=maxi[q]+1;            else            {                int k=tail++;                fail[k]=fail[q];                fail[q]=fail[t]=k;                maxi[k]=maxi[p]+1;                mini[q]=maxi[k]+1;                mini[t]=maxi[k]+1;                for(int i=0;i<26;i++)                    next[k][i]=next[q][i];                while(p&&next[p][c]==q)                    next[p][c]=k,p=fail[p];                mini[k]=maxi[fail[k]]+1;            }        }        else fail[t]=1,mini[t]=1;        return t;    }}using namespace suffixAuttomaton;int n;char s[mxn];int main(){    scanf("%s",s);    n=strlen(s);    int last=1;    for(int i=0;i<n;i++)        last=extend(last,s[i]-'a');    long long ans=0;    for(int i=2;i<tail;i++)        ans+=maxi[i]-mini[i]+1;    printf("%lld\n",ans );}
阅读全文
0 0
原创粉丝点击