码迷,mamicode.com
首页 > 其他好文 > 详细

后缀自动机

时间:2020-02-07 20:56:24      阅读:65      评论:0      收藏:0      [点我收藏+]

标签:long   pac   alt   cout   cli   play   else   amp   span   

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

 

后缀自动机

标签:long   pac   alt   cout   cli   play   else   amp   span   

原文地址:https://www.cnblogs.com/bxd123/p/12274269.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!