洛谷 P3804 后缀自动机

Posted 蒟蒻JHY

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了洛谷 P3804 后缀自动机相关的知识,希望对你有一定的参考价值。

题目描述

给定一个只包含小写字母的字符串SS ,

请你求出 SS 的所有出现次数不为 11 的子串的出现次数乘上该子串长度的最大值。

输入输出格式

输入格式:

 

一行一个仅包含小写字母的字符串SS

 

输出格式:

 

一个整数,为 所求答案

 

输入输出样例

输入样例#1: 
abab
输出样例#1: 
4

说明

对于10\%10% 的数据,|S|<=1000S<=1000

对于100\%100% 的数据,|S|<=10^6S<=106

 

今天上午恶补了一下后缀自动机,发现好像没有想象中的那么难,,,,

对于本题来说,我们只要把后缀自动机建出来,然后逆序dfs维护一下非根非叶子的right集合大小*max{}的最大值即可。

 

#include<bits/stdc++.h>
#define ll long long
#define maxn 4000005
#define pb push_back
using namespace std;
int root=1,tot=1,n;
char s[maxn];
int f[maxn],ch[maxn][27];
int siz[maxn],las=1,l[maxn];
int a[maxn],c[maxn];
ll ans=0;

inline void ins(int x){
    int p=las,np=++tot;
    las=np,l[np]=l[p]+1;
    siz[np]=1;
    
    for(;p&&!ch[p][x];p=f[p]) ch[p][x]=np;
    if(!p) f[np]=1;
    else{
        int q=ch[p][x];
        if(l[q]==l[p]+1) f[np]=q;
        else{
            int nq=++tot;
            l[nq]=l[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            f[nq]=f[q];
            f[q]=f[np]=nq;
            for(;ch[p][x]==q;p=f[p]) ch[p][x]=nq;
        }
    }
}

inline void build(){
    for(int i=0;i<n;i++) ins(s[i]-‘a‘);
    for(int i=1;i<=tot;i++) c[l[i]]++;
    for(int i=n;i>=0;i--) c[i]+=c[i+1];
    for(int i=1;i<=tot;i++) a[c[l[i]]--]=i;
}

inline void solve(){
    for(int i=1;i<=tot;i++){
        int now=a[i];
        siz[f[now]]+=siz[now];
        if(siz[now]>1&&siz[now]<n) ans=max(ans,siz[now]*(ll)l[now]);
    }
}

int main(){
    scanf("%s",s);
    n=strlen(s);
    build();
    solve();
    printf("%lld\n",ans);
    return 0;
}

  

以上是关于洛谷 P3804 后缀自动机的主要内容,如果未能解决你的问题,请参考以下文章

洛谷 P3804 后缀自动机

P3804 模板后缀自动机

P3804 后缀自动机模版

P3804 模板后缀自动机

后缀自动机

109 后缀自动机(SAM)