Luogu3804：[模板]后缀自动机

题面

luogu

Sol

$sam$然后树形$DP$
当时还不会拓扑排序的我

# include <bits/stdc++.h>
# define IL inline
# define RG register
# define Fill(a, b) memset(a, b, sizeof(a))
using namespace std;
typedef long long ll;

template <class Int>
IL void Input(RG Int &x){
    RG int z = 1; RG char c = getchar(); x = 0;
    for(; c < '0' || c > '9'; c = getchar()) z = c == '-' ? -1 : 1;
    for(; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
    x *= z;
}

const int maxn(2e6 + 5);

int n, trans[26][maxn], fa[maxn], len[maxn], tot = 1, last = 1, size[maxn];
ll ans;
vector <int> edge[maxn];
char s[maxn];

IL void Extend(RG int c){
    RG int np = ++tot, p = last; last = np;
    len[np] = len[p] + 1, size[np] = 1;
    while(p && !trans[c][p]) trans[c][p] = np, p = fa[p];
    if(!p) fa[np] = 1;
    else{
        RG int q = trans[c][p];
        if(len[p] + 1 == len[q]) fa[np] = q;
        else{
            RG int nq = ++tot;
            len[nq] = len[p] + 1, fa[nq] = fa[q];
            for(RG int i = 0; i < 26; ++i) trans[i][nq] = trans[i][q];
            fa[q] = fa[np] = nq;
            while(p && trans[c][p] == q) trans[c][p] = nq, p = fa[p];
        }
    }
}

IL void Dfs(RG int x){
    for(RG int l = edge[x].size(), e = 0; e < l; ++e)
        Dfs(edge[x][e]), size[x] += size[edge[x][e]];
    if(size[x] != 1) ans = max(ans, 1LL * size[x] * len[x]);
}

int main(RG int argc, RG char* argv[]){
    scanf(" %s", s), n = strlen(s);
    for(RG int i = 0; i < n; ++i) Extend(s[i] - 'a');
    for(RG int i = 2; i <= tot; ++i) edge[fa[i]].push_back(i);
    Dfs(1);
    printf("%lld\n", ans);
    return 0;
}