洛谷P3804（SAM）

題目鏈接：https://www.luogu.org/problem/P3804

思路：sam模板

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <unordered_map>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%lld",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(998244353)
#define pb push_back
#define eps 1e-7
#define lc d<<1
#define rc d<<1|1
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
const int MAXN=1e6+8;
char a[MAXN];
int sum[MAXN<<1],t[MAXN<<1];
ll dp[MAXN<<1],ans;
struct SAM{
    int las=1,tot=1,ch[MAXN<<1][26],len[MAXN<<1],fa[MAXN<<1];
    inline void add(int c)
    {
        int p=las;int np=las=++tot;dp[tot]=1;
        len[np]=len[p]+1;
        for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
        if(!p) fa[np]=1;//以上爲case 1
        else
        {
            int q=ch[p][c];
            if(len[q]==len[p]+1) fa[np]=q;//以上爲case 2
            else
            {
                int nq=++tot;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                len[nq]=len[p]+1;
                fa[nq]=fa[q];
                fa[q]=fa[np]=nq;
                for(;p&&ch[p][c]==q;p=fa[p]) ch[p][c]=nq;//以上爲case 3
            }
        }
    }
    inline void slove()
    {
        FOR(i,1,tot) sum[len[i]]++;
        FOR(i,1,tot) sum[i]+=sum[i-1];
        FOR(i,1,tot) t[sum[len[i]]--]=i;
        FOL(i,tot,1)
        {
            int v=t[i];
            dp[fa[v]]+=dp[v];
            if(dp[v]>1) ans=max(ans,dp[v]*len[v]);
        }
        cout<<ans<<endl;
    }
}sam;
int main()
{
    cin.tie(0);
    cout.tie(0);
    ss(a);int n=strlen(a);
    for(int i=0;i<n;i++) sam.add(a[i]-'a');
    sam.slove();
    return 0;
}