后缀自动机模板——不同子串个数p2408

后缀自动机的入门博客 https://www.luogu.org/blog/Kesdiael3/hou-zhui-zi-dong-ji-yang-xie

有两种求法，分别对应了两种性质

#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
char s[maxn];
int n;

int cmp(int a,int b);
struct SAM{
    int cnt,last;
    int nxt[maxn][];
    int link[maxn];
    int len[maxn];
    long long ans[maxn];
    SAM(){
        cnt=last=;
    }
    void insert(int c){
        int p=last,np=last=++cnt;
        len[np]=len[p]+;
        for(;p&&!nxt[p][c];p=link[p])
            nxt[p][c]=np;
        if(!p) {link[np]=;return;}

        int q=nxt[p][c];
        if(len[q]==len[p]+){link[np]=q;return;}

        int clone=++cnt;
        link[clone]=link[q];
        len[clone]=len[p]+;
        memcpy(nxt[clone],nxt[q],sizeof nxt[q]);
        link[q]=link[np]=clone;
        for(;p&&nxt[p][c]==q;p=link[p])
            nxt[p][c]=clone;
    }
    int id[maxn<<];
    void calc(){
        for(int i=;i<=cnt;i++)id[i]=i;
        sort(id+,id+cnt+,cmp);//按状态所对应的长度排序
        for(int i=;i<=cnt;i++)
            for(int j=;j<;j++)
                if(nxt[id[i]][j])
                    ans[id[i]]+=ans[nxt[id[i]][j]]+;
    }
    void calc2(){//这里两种写法都可以
        for(int i=;i<=cnt;i++)
            ans[]+=len[i]-len[link[i]];
    }
}p;
int cmp(int a,int b){return p.len[a]>p.len[b];}

int main(){
    scanf("%d%s",&n,s);
    for(int i=;i<n;i++)
        p.insert(s[i]-'a');
    p.calc();
    cout<<p.ans[]<<endl;
}