后缀自动机

Posted 2021-03-11 bxd123

P3804 【模板】后缀自动机 (SAM)

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+100;
typedef long long ll;

char s[N];
int ch[N*2][26],len[N*2],fa[N*2];
int last=1,tot=1;
int cnt[N],c[N];
void add(int c) {
    int p=last,np=last=++tot;
    len[np]=len[p]+1; cnt[np]=1;
    for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=1;
    else {
        int q=ch[p][c];
        if(len[q]==len[p]+1) fa[np]=q;
        else {
            int nq=++tot;len[nq]=len[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            for(;p&&ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
        }
    }
}
int a[N];
ll ans;
int main() {
    cin>>s+1;
    for(int i=1;s[i];i++) add(s[i]-‘a‘);
    for(int i=1;i<=tot;i++) c[len[i]]++;
    for(int i=1;i<=tot;i++) c[i]+=c[i-1];
    for(int i=1;i<=tot;i++) a[c[len[i]]--]=i;
    for(int i=tot;i;i--) {
        int p=a[i];
        cnt[fa[p]]+=cnt[p];
        if(cnt[p]>1) ans=max(ans,1ll*cnt[p]*len[p]);
    }
    cout<<ans;
}

View Code

 

 

P3181 [HAOI2016]找相同字符

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+100;
typedef long long ll;

char s[N];
int ch[N*2][27],len[N*2],fa[N*2];
int last=1,tot=1;
int cnt[N][2],c[N];
void add(int c,int f) {
    int p=last,np=last=++tot;
    len[np]=len[p]+1; cnt[np][f]=1;
    for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=1;
    else {
        int q=ch[p][c];
        if(len[q]==len[p]+1) fa[np]=q;
        else {
            int nq=++tot;len[nq]=len[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            for(;p&&ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
        }
    }
}
int a[N],le;
ll ans;
int main() {
    cin>>s+1;le=strlen(s+1);
    for(int i=1;i<=le;i++) add(s[i]-‘a‘,0);
    add(26,0);
    cin>>s+1;le=strlen(s+1);
    for(int i=1;i<=le;i++) add(s[i]-‘a‘,1);
    for(int i=1;i<=tot;i++) c[len[i]]++;
    for(int i=1;i<=tot;i++) c[i]+=c[i-1];
    for(int i=1;i<=tot;i++) a[c[len[i]]--]=i;
    for(int i=tot;i;i--) {
        int p=a[i];
        cnt[fa[p]][0]+=cnt[p][0];
        cnt[fa[p]][1]+=cnt[p][1];
    }
    for(int i=1;i<=tot;i++) ans+=1ll*cnt[i][0]*cnt[i][1]*(len[i]-len[fa[i]]);
    cout<<ans;
}

View Code