bzoj4567: [Scoi2016]背单词【trie树+贪心】

Description

Lweb 面对如山的英语单词，陷入了深深的沉思，“我怎么样才能快点学完，然后去玩三国杀呢？”。这时候睿智
的凤老师从远处飘来，他送给了 Lweb 一本计划册和一大缸泡椒，他的计划册是长这样的：
—————
序号 单词
—————
1
2
……
n-2
n-1
n
—————
然后凤老师告诉 Lweb ，我知道你要学习的单词总共有 n 个，现在我们从上往下完成计划表，对于一个序号为 x
的单词（序号 1…x-1 都已经被填入）：
1) 如果存在一个单词是它的后缀，并且当前没有被填入表内，那他需要吃 n×n 颗泡椒才能学会；
2) 当它的所有后缀都被填入表内的情况下，如果在 1…x-1 的位置上的单词都不是它的后缀，那么你吃 x 颗泡
椒就能记住它；
3) 当它的所有后缀都被填入表内的情况下，如果 1…x-1的位置上存在是它后缀的单词，所有是它后缀的单词中
，序号最大为 y ，那么你只要吃 x-y 颗泡椒就能把它记住。
Lweb 是一个吃到辣辣的东西会暴走的奇怪小朋友，所以请你帮助 Lweb ，寻找一种最优的填写单词方案，使得他
记住这 n 个单词的情况下，吃最少的泡椒。

Input

输入一个整数 n ，表示 Lweb 要学习的单词数。接下来 n 行，每行有一个单词（由小写字母构成，且保证任意单
词两两互不相同）1≤n≤100000, 所有字符的长度总和 1≤|len|≤510000
Output

Lweb 吃的最少泡椒数

Sample Input

2
a
ba

Sample Output

2

解题思路：

看来没读懂题的不止我一个。

意思就是挨个放单词，放每个单词之前一定把这个单词的后缀都先放上去，每个单词的代价等于这个单词的位置减去上一个出现的这个单词的后缀的位置。

如果我们把所有串反过来建一棵trie树，在把结尾节点构成的树结构提出来组成一棵新树，那么问题就变成了求一个最优的遍历方式，使得每个点的访问序-父亲的访问序之和最小。

结论是每次贪心优先遍历size最小的子树。

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int N=510005;
struct node{int son[26],ed;}tr[N];
int n,tot,idx,dfn[N],size[N];
char s[N];ll ans;
vector<int>g[N];
inline bool cmp(const int &a,const int &b){return size[a]<size[b];}
void Insert()
{
    int len=strlen(s),po=0;
    for(int i=len-1;i>=0;i--)
    {
        if(!tr[po].son[s[i]-'a'])tr[po].son[s[i]-'a']=++tot;
        po=tr[po].son[s[i]-'a'];
    }
    tr[po].ed=1;
}
void dfs1(int u,int fa)
{
    if(tr[u].ed)size[u]=1,g[fa].push_back(u);
    for(int i=0;i<26;i++)
    {
        int v=tr[u].son[i];
        if(!v)continue;
        dfs1(v,tr[u].ed?u:fa);
        size[u]+=size[v];
    }
}
void dfs2(int u,int fa)
{
    dfn[u]=++idx,ans+=dfn[u]-dfn[fa];
    for(int i=0;i<g[u].size();i++)
        dfs2(g[u][i],u);
}
int main()
{
    //freopen("lx.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%s",s),Insert();
    dfs1(0,0);
    for(int i=0;i<=tot;i++)sort(g[i].begin(),g[i].end(),cmp);
    dfs2(0,0);
    cout<<ans<<'\n';
    return 0;
}