【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[N][30],size[N],fa[N],n,m,cnt,deep[N],id;
long long ans; 
char c[N];
vector<pair<int,int> >s[N];
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; 
}