HihoCoder - 1445 后缀自动机 试水题

题意:求子串个数

SAM中每个子串包含于某一个状态中

对于不同的状态\(u,v\),\(sub(u)∩sub(v)=NULL\)

因此答案就是对于所有的状态\(st\),\(ans=\sum_{st} maxlen(st)-minlen(st)+1\)

#include<bits/stdc++.h>
#include<unordered_map>
#define rep(i,j,k) for(register int i=j;i<=k;i++)
#define rrep(i,j,k) for(register int i=j;i>=k;i--)
#define erep(i,u) for(register int i=head[u];~i;i=nxt[i])
#define fastIO ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define println(x) print(x),print('\n')
using namespace std;
using ll = long long;
using pr = pair<ll,ll>;
const int MAXN = 1e6+11;
const int INF  = 0x3f3f3f3f;
const double EPS = 1e-7;

struct SAM{
    int tot;
    int trans[MAXN<<1][26];
    int slink[MAXN<<1];
    int maxlen[MAXN<<1],minlen[MAXN<<1];
    void init(){
        tot=1;
        maxlen[1]=minlen[1]=slink[1]=0;
        rep(i,0,25) trans[1][i]=0;
    }
    int node(int mx,int mn,int old,int _slink){
        ++tot;
        maxlen[tot]=mx;
        minlen[tot]=mn;
        memcpy(trans[tot],trans[old],sizeof trans[0]);
        slink[tot]=_slink;
        return tot;
    }
    int add(char ch,int u){
        int c=ch-'a';
        int z=node(maxlen[u]+1,0,0,0);
        int v=u;
        while(v&&trans[v][c]==0){
            trans[v][c]=z;
            v=slink[v];
        }
        if(v==0){
            minlen[z]=1;
            slink[z]=1;
            return z;
        }
        int x=trans[v][c];//compare
        if(maxlen[x]==maxlen[v]+1){
            slink[z]=x;
            minlen[z]=maxlen[x]+1;
            return z;
        }
        int y=node(maxlen[v]+1,0,x,slink[x]);//chai dian
        slink[y]=slink[x]; //ji cheng
        slink[x]=y;  //chai dian y laizi yu x
        slink[z]=y;  //fen pei z de suffix-path 

        minlen[x]=maxlen[v]+2; //=maxlen[y]+1
        minlen[z]=maxlen[y]+1;

        int w=v;
        while(w&&trans[w][c]==x){
            trans[w][c]=y;
            w=slink[w];
        }

        minlen[y]=maxlen[slink[y]]+1; //biran kebujilu

        return z;
    }
}sam;
char str[MAXN];
int main(){
    #ifndef ONLINE_JUDGE
    freopen("stdin.txt","r",stdin);
    #endif
    while(~scanf("%s",str+1)){
        sam.init();
        int last=1;
        int len=strlen(str+1);
        for(int i=1;i<=len;i++){
            last=sam.add(str[i],last);
        }
        ll ans=0;
        for(int i=2;i<=sam.tot;i++){
            ans+=sam.maxlen[i]-sam.minlen[i]+1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}