[BZOJ4567][Scoi2016]背单词 字典树+贪心

后缀关系可以构成一棵树，这东西用字典树就能求

那么等同于说需要把这棵树上的节点放到序列上，显然不可能把父亲放在儿子后面，$n^2$的代价大于任何一种按照dfs序放的代价

显然放到序列上是按照dfs序来放的，这个推一推就推出来了
dfs优先走siz小的儿子（这个和树剖刚好相反），正确性显然（小学数学书好像有类似的结论）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>

#define N 510050
using namespace std;
vector<int> e[N];
char s[N];
int fa[N],siz[N],dfn[N],tme,n;
bool vis[N];
long long ans;
int ch[N][27],leef[N],l,f,p,cnt,tot;
void add() {
    l = strlen(s+1) , f = p = 0;
    for (int _=l;_>=1;_--) {
        int c = s[_] - 'a' + 1;
        if (!ch[p][c]) ch[p][c] = ++cnt;
        p = ch[p][c];
    }
    leef[p] = ++tot;
}

void dfs_trie(int x,int f) {
    if (leef[x]) fa[ leef[x] ] = f;
    for (int i=1;i<=26;i++) if (ch[x][i])
        dfs_trie(ch[x][i],leef[x]?leef[x]:f);
}

void dfs1(int x) {
    siz[x] = 1;
    for (int i=0;i<(int)e[x].size();i++) {
        int y = e[x][i];
        dfs1(y); siz[x] += siz[y];
    }
}

bool cmp(int p1,int p2) { return siz[p1] < siz[p2]; }
void dfs2(int x) {
    dfn[x]=++tme; ans += 1LL * dfn[x] - dfn[ fa[x] ];
    sort(e[x].begin(),e[x].end(),cmp);
    for (int i=0;i<(int)e[x].size();i++) 
        dfs2(e[x][i]);
}

int main() {
    #ifndef ONLINE_JUDGE
        freopen("1.in","r",stdin);
    #endif
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%s",s+1) , add();
    dfs_trie(0,0);
    for (int i=1;i<=n;i++) e[ fa[i] ].push_back(i);
    dfs1(0);
    dfs2(0);
    cout << ans << endl;
    return 0;
}