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

Posted 2022-06-15 zsben991126

后缀自动机的入门博客 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][26];
    int link[maxn];
    int len[maxn];
    long long ans[maxn];
    SAM()
        cnt=last=1;
    
    void insert(int c)
        int p=last,np=last=++cnt;
        len[np]=len[p]+1;
        for(;p&&!nxt[p][c];p=link[p])
            nxt[p][c]=np;
        if(!p) link[np]=1;return;
        
        int q=nxt[p][c];
        if(len[q]==len[p]+1)link[np]=q;return;
        
        int clone=++cnt;
        link[clone]=link[q];
        len[clone]=len[p]+1;
        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<<1];
    void calc()
        for(int i=1;i<=cnt;i++)id[i]=i;
        sort(id+1,id+cnt+1,cmp);//按状态所对应的长度排序 
        for(int i=1;i<=cnt;i++)
            for(int j=0;j<26;j++)
                if(nxt[id[i]][j])
                    ans[id[i]]+=ans[nxt[id[i]][j]]+1;
    
    void calc2()//这里两种写法都可以 
        for(int i=1;i<=cnt;i++)
            ans[1]+=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=0;i<n;i++)
        p.insert(s[i]-‘a‘);
    p.calc();
    cout<<p.ans[1]<<endl;