51Nod-1469 淋漓盡致子串 (後綴自動機)

51Nod-1469 淋漓盡致子串 (後綴自動機)

對於原串構造後綴自動機

咱們只考慮\(|endpos|>1\)的狀態

發現對於一個狀態，只有其中最長的子串會產生貢獻，不然會在\(p_i-1\)出產生衝突

對於最長串的\(p_i-1\)位置，若是\(link/parent\)樹上的一個兒子的\(|endpos|>1\)，那麼意味着存在衝突（由於兒子就是在串前面添加一些字符獲得的）

對於\(p_i+len\)的狀況，實際上就是\(j\in endpos,j+1\)不能產生衝突，在\(link,parent\)樹上進行子樹統計便可

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

#define pb push_back
template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); }

char IO;
int rd(){
    int s=0;
    int f=0;
    while(!isdigit(IO=getchar())) if(IO=='-') f=1;
    do s=(s<<1)+(s<<3)+(IO^'0');
    while(isdigit(IO=getchar()));
    return f?-s:s;
}

const int N=2e5+10,P=1e9+7;

int n;
char s[N];
int trans[N][26],link[N],len[N],lst,stcnt,End[N];
int sz[N];

struct Edge{
    int to,nxt;
}e[N];
int head[N],ecnt;
void AddEdge(int u,int v) {
    e[++ecnt]=(Edge){v,head[u]};
    head[u]=ecnt;
}

void Init(){
    link[0]=-1,len[0]=0;
    rep(i,0,stcnt) rep(j,0,25) trans[i][j]=0;
    stcnt=lst=0;
}

void Extend(int c) {
    int cur=++stcnt,p=lst;
    End[cur]=len[cur]=len[p]+1,sz[cur]=1;
    while(~p && !trans[p][c]) trans[p][c]=cur,p=link[p];
    if(p==-1) link[cur]=0;
    else {
        int q=trans[p][c];
        if(len[q]==len[p]+1) link[cur]=q;
        else {
            int clone=++stcnt;
            memcpy(trans[clone],trans[q],104);
            link[clone]=link[q],len[clone]=len[p]+1;
            while(~p && trans[p][c]==q) trans[p][c]=clone,p=link[p];
            link[cur]=link[q]=clone;
        }
    }
    lst=cur;
}

int ans;
int apr[N][26];
void dfs(int u) {
    if(End[u] && End[u]<n) apr[u][s[End[u]+1]-'a']=1;
    int fl=1;
    for(reg int i=head[u];i;i=e[i].nxt) {
        int v=e[i].to;
        dfs(v);
        if(sz[v]>1) fl=0; //統計兒子沒有sz>1
        rep(j,0,25) apr[u][j]+=apr[v][j]; // 子樹統計endpos+1處的衝突
        sz[u]+=sz[v];
    }
    if(u && sz[u]>1) { //sizeof endpos>1
        rep(i,0,25) if(apr[u][i]>1) fl=0; // 衝突
        ans+=fl;//只考慮最長的串
    }
}

int main(){
    scanf("%s",s+1),n=strlen(s+1);
    Init();
    rep(i,1,n) Extend(s[i]-'a');
    rep(i,1,stcnt) AddEdge(link[i],i);
    dfs(0);
    printf("%d\n",ans);
}