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=1000010;
int tot=1,n,slink[2*N],trans[2*N][26],minlen[2*N],maxlen[2*N];
char str[N];
ll ans=0;
int newstate(int _maxlen,int _minlen,int* _trans,int _slink) {
    maxlen[++tot]=_maxlen;
    minlen[tot]=_minlen;
    slink[tot]=_slink;
    if(_trans)
        for(int i=0; i<26; i++)
            trans[tot][i]=_trans[i];
    else for(int i=0;i<26;i++)
            trans[tot][i]=-1;
    return tot;
}
int add_char(char ch,int u) {
    int c=ch-'a',v=u;
    int z=newstate(maxlen[u]+1,-1,NULL,0);
    while(v!=-1&&trans[v][c]==-1) {
        trans[v][c]=z;
        v=slink[v];
    }
    if(v==-1) {
        minlen[z]=1;
        slink[z]=0;
        return z;
    } 
    int x=trans[v][c];
    if(maxlen[v]+1==maxlen[x]) {
        slink[z]=x;
        minlen[z]=maxlen[x]+1;
        return z;
    }
    int y=newstate(maxlen[v]+1,-1,trans[x],slink[x]);
    slink[z]=slink[x]=y;
    minlen[x]=minlen[z]=maxlen[y]+1;
    while(v!=-1&&trans[v][c]==x) {
        trans[v][c]=y;
        v=slink[v];
    }
    minlen[y]=maxlen[slink[y]]+1;
    return z;
}
int main() {
    scanf("%s",str);
    int len=strlen(str),pre=0;
    memset(trans[0],-1,sizeof(trans[0]));
    slink[0]=-1;
    for(int i=0; i<len; i++) {
        pre=add_char(str[i],pre);
    }
    for(int i=2; i<=tot; i++) {
        ans+=maxlen[i]-minlen[i]+1;
    }
    printf("%lld\n",ans);
    return 0;
}

 

做了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=1000100;
int tot,slink[2*maxn],trans[2*maxn][26],maxlen[2*maxn];
char str[2*maxn];
int N,M,K,num[2*maxn],last;
long long ans;
void init()
{
    ans=tot=0;
    last=0;
    memset(trans[0],-1,sizeof(trans[0]));
    slink[0]=-1; maxlen[0]=0;
}
void add_char(char chr) 
{
    int c=chr-'a';
    int p=last,np=++tot;
    maxlen[np]=maxlen[p]+1;
    memset(trans[np],-1,sizeof(trans[np])); 
    num[np]=0;
    while(p!=-1&&trans[p][c]==-1)  trans[p][c]=np,p=slink[p];
    if(p==-1) slink[np]=0;
    else
    {
        int q=trans[p][c];
        if(maxlen[q]!=maxlen[p]+1)
        {
            int nq=++tot;
            memcpy(trans[nq],trans[q],sizeof(trans[q])); num[nq]=num[q];
            maxlen[nq]=maxlen[p]+1;
            slink[nq]=slink[q];
            slink[np]=slink[q]=nq;
            while(p!=-1&&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=0; i<N; i++)   add_char(str[i]);
    for(int i=1;i<=tot;i++) ans+=maxlen[i]-maxlen[slink[i]];
    printf("%lld\n",ans);
    return 0;
}