后缀自动机 求不同的子串数目

时间限制:10000ms
单点时限:2000ms
内存限制:512MB
描述

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

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

#include<iostream>
#include<cstdio>
#include<cstring>
#define rg register
#define rep(i,x,y) for(rg int i=(x);i<=(y);++i)
#define per(i,x,y) for(rg int i=(x);i>=(y);--i)
using namespace std;
typedef long long LL;
const int N=1e6+10,P=N<<1,alp=26;
inline bool vaild(char c){return c>='a'&&c<='z';}
inline char gc(){char c=getchar();while(!vaild(c))c=getchar();return c;}
int n,s[N];
struct SAM{
    int son[P][alp],pr[P],step[P],tot,last;
    SAM(){last=tot=1;}
    void ins(int c,int l){
        int k=++tot;step[k]=l;
        while(last&&son[last][c]==0)son[last][c]=k,last=pr[last];
        if(last){
            int q=son[last][c];
            if(step[q]==step[last]+1)pr[k]=q;
            else{
                int nq=++tot;
                step[nq]=step[last]+1;
                memcpy(son[nq],son[q],sizeof(son[q]));
                pr[nq]=pr[q];
                pr[q]=pr[k]=nq;
                while(last&&son[last][c]==q)son[last][c]=nq,last=pr[last];
            }
        }else pr[k]=1;
        last=k;
    }
    void build(int*a,int n){
        rep(i,1,n)ins(a[i],i);
    }
    LL count(){
        LL ret=0;
        rep(i,1,tot)ret+=step[i]-step[pr[i]];
        return ret;
    }
}sam;
int main(){
    char c=gc();
    while(vaild(c))s[++n]=c-'a',c=getchar();
    sam.build(s,n);
    printf("%lld",sam.count());
    return 0;
}