【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.最后答案一定是这棵trie的某个dfs序.

用size[x]代表x这棵子树里的单词个数.分析一下可以发现应该先走size大的节点.

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define N 550000
using namespace std;
int ch
[30],size
,fa
,n,m,cnt,deep
,id;
long long ans;
char c
;
vector<pair<int,int> >s
;
void insert(){
scanf("%s",c);
int len=strlen(c),now=0;
for (int i=len-1;i>=0;i--){
int t=c[i]-'a'+1;
if (!ch[now][t]) ch[now][t]=++cnt;
now=ch[now][t];
}
size[now]=1;
}
void dfs(int x,int ff){
if (size[x]){fa[x]=ff;ff=x;}
for (int i=1;i<=26;i++)
if (ch[x][i]) dfs(ch[x][i],ff);
}
void dfs2(int x,int ff){
deep[x]=++id;
if (size[x]) ans+=(long long)(deep[x]-deep[ff]);
for (int i=0;i<s[x].size();i++)
dfs2(s[x][i].second,x);
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) insert();
dfs(0,0);
for (int i=cnt;i;i--)
if (size[i]){size[fa[i]]+=size[i];s[fa[i]].push_back(make_pair(size[i],i));}
for (int i=0;i<=cnt;i++)
if (size[i]) sort(s[i].begin(),s[i].end());
id=-1;dfs2(0,0);
cout<<ans<<endl;
}