后缀自动机的入门博客 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; }
原文地址:https://www.cnblogs.com/zsben991126/p/11309136.html
时间: 2024-10-13 01:04:51