洛谷 P3804 后缀自动机
题目描述
给定一个只包含小写字母的字符串SS ,
请你求出 SS 的所有出现次数不为 11 的子串的出现次数乘上该子串长度的最大值。
输入输出格式
输入格式:
一行一个仅包含小写字母的字符串SS
输出格式:
一个整数,为 所求答案
输入输出样例
说明
对于10\%10% 的数据,|S|<=1000∣S∣<=1000
对于100\%100% 的数据,|S|<=10^6∣S∣<=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; }
我爱学习,学习使我快乐