HihoCoder1445 重复旋律5（后缀自动机）

重复旋律5

时间限制:10000ms

单点时限:2000ms

内存限制:512MB

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为一段数构成的数列。

现在小Hi想知道一部作品中出现了多少不同的旋律？

输入

共一行，包含一个由小写字母构成的字符串。字符串长度不超过 1000000。

输出

一行一个整数，表示答案。

样例输入

aab

样例输出

5

2017-11-22：从代码看，是比ac自动机优美，但是ac自动机是基础在kmp和字典树上面，学习起来很快，然而后缀自动机就灰常难以YY啦。

目前感觉似懂非懂，等把后面几个题AC了再回来整理吧。

 

2017-11-25：三天，做了几个题之后，基本上是弄懂了。感觉后缀自动机很强大，打算再花一周来练习。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=;
int tot=,n,slink[*N],trans[*N][],minlen[*N],maxlen[*N];
char str[N];
ll ans=;
int newstate(int _maxlen,int _minlen,int* _trans,int _slink) {
    maxlen[++tot]=_maxlen;
    minlen[tot]=_minlen;
    slink[tot]=_slink;
    if(_trans)
        for(int i=; i<; i++)
            trans[tot][i]=_trans[i];
    else for(int i=;i<;i++)
            trans[tot][i]=-;
    return tot;
}
int add_char(char ch,int u) {
    int c=ch-'a',v=u;
    int z=newstate(maxlen[u]+,-,NULL,);
    while(v!=-&&trans[v][c]==-) {
        trans[v][c]=z;
        v=slink[v];
    }
    if(v==-) {
        minlen[z]=;
        slink[z]=;
        return z;
    }
    int x=trans[v][c];
    if(maxlen[v]+==maxlen[x]) {
        slink[z]=x;
        minlen[z]=maxlen[x]+;
        return z;
    }
    int y=newstate(maxlen[v]+,-,trans[x],slink[x]);
    slink[z]=slink[x]=y;
    minlen[x]=minlen[z]=maxlen[y]+;
    while(v!=-&&trans[v][c]==x) {
        trans[v][c]=y;
        v=slink[v];
    }
    minlen[y]=maxlen[slink[y]]+;
    return z;
}
int main() {
    scanf("%s",str);
    int len=strlen(str),pre=;
    memset(trans[],-,sizeof(trans[]));
    slink[]=-;
    for(int i=; i<len; i++) {
        pre=add_char(str[i],pre);
    }
    for(int i=; i<=tot; i++) {
        ans+=maxlen[i]-minlen[i]+;
    }
    printf("%lld\n",ans);
    return ;
}

做了HDU4641再回来改动一下，minlen[i]=maxlen[slink[i]],所以没必要保存minlen了。代码稍微简短了一点：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
using namespace std;
const int maxn=;
int tot,slink[*maxn],trans[*maxn][],maxlen[*maxn];
char str[*maxn];
int N,M,K,num[*maxn],last;
long long ans;
void init()
{
    ans=tot=;
    last=;
    memset(trans[],-,sizeof(trans[]));
    slink[]=-; maxlen[]=;
}
void add_char(char chr)
{
    int c=chr-'a';
    int p=last,np=++tot;
    maxlen[np]=maxlen[p]+;
    memset(trans[np],-,sizeof(trans[np]));
    num[np]=;
    while(p!=-&&trans[p][c]==-)  trans[p][c]=np,p=slink[p];
    if(p==-) slink[np]=;
    else
    {
        int q=trans[p][c];
        if(maxlen[q]!=maxlen[p]+)
        {
            int nq=++tot;
            memcpy(trans[nq],trans[q],sizeof(trans[q])); num[nq]=num[q];
            maxlen[nq]=maxlen[p]+;
            slink[nq]=slink[q];
            slink[np]=slink[q]=nq;
            while(p!=-&&trans[p][c]==q) trans[p][c]=nq,p=slink[p];
        }
        else slink[np]=q;
    }
    last=np;
}
int main() {
    init();
    scanf("%s",str);
    N=strlen(str);
    for(int i=; i<N; i++)   add_char(str[i]);
    for(int i=;i<=tot;i++) ans+=maxlen[i]-maxlen[slink[i]];
    printf("%lld\n",ans);
    return ;
}