[SCOI2016]背单词(trie+贪心）

#1459(校内OJ) [SCOI2016]背单词

题意描述部分//

描述

Lweb 面对如山的英语单词，陷入了深深的沉思，”我怎么样才能快点学完，然后去玩三国杀呢？“。这时候睿智的凤老师从远处飘来，他送给了 Lweb 一本计划册和一大缸泡椒，他的计划册是长这样的：
—————序号 单词—————
1 2…n-2n-1 n—————
然后凤老师告诉 Lweb ，我知道你要学习的单词总共有 n 个，现在我们从上往下完成计划表，对于一个序号为 x 的单词（序号 1…x-1 都已经被填入）：

如果存在一个单词是它的后缀，并且当前没有被填入表内，那他需要吃 n*n 颗泡椒才能学会；

当它的所有后缀都被填入表内的情况下，如果在 1…x-1 的位置上的单词都不是它的后缀，那么你吃 x 颗泡椒就能记住它；

当它的所有后缀都被填入表内的情况下，如果 1…x-1的位置上存在是它后缀的单词，所有是它后缀的单词中，序号最大为 y ，那么你只要吃 x-y 颗泡椒就能把它记住。

Lweb 是一个吃到辣辣的东西会暴走的奇怪小朋友，所以请你帮助 Lweb ，寻找一种最优的填写单词方案，使得他记住这 n 个单词的情况下，吃最少的泡椒。

题解如下///

1.看到后缀匹配显然转成trie
2.读完题，发现二三条件都是在避免一的前提下进行的，根据贪心方法不难想到让每个单词都在前缀之后；
3.根据第三点，显然把一个前缀后面的单词放在一起；

单词由前缀前缀一次连接下去，最后是一棵树；

最后一个问题，不同的前缀子树如何排序？

根据第二个，不难发现，每一个前缀子树内部的权值都是相对独立的，当前只需要每一个子树的最前缀所对应权值和最小；
所以显然把节点少的放在 前面；这样就有排序标准了；

考虑dfs实现；(一个单词可能有前缀，也成为后面单词的前缀，所以一个前缀子树内的情况是一样的，递归实现）

（代码注： dfs中计算当前答案：第i大的节点数会被算 （k-i）次，k是当前考虑的子树总数）

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define in red()
inline ll red()
{
    int data=0;int w=1; char ch=0;
    ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data*w;
}
const int maxn=3e5+10;
int f[maxn][30],cnt=0;
ll w[maxn];
bool flag[maxn];
vector <int> G[maxn];
struct edge{
 int nxt,v;
}e[maxn];
int head[maxn],ecnt=0;
inline void _add(int u,int v){
 e[++ecnt].v=v,e[ecnt].nxt=head[u],head[u]=ecnt;
}
ll ans=0;
inline void insert(char s[]){
 int u=0;
 for(int i=strlen(s)-1;i>=0;i--){
  int t=s[i]-'a';
  if(!f[u][t])f[u][t]=++cnt,G[u].push_back(t);
  u=f[u][t];
  //前缀标记有问题
  //重新走一遍 
 }
 flag[u]=1;
 // 要记录自己 
}inline void rego(int u,int fa){
 int las=(flag[u]) ? u:fa;
 for(int i=0;i<G[u].size();i++){
  int v=f[u][G[u][i]];
  if(flag[v])_add(las,v);
  rego(v,las);
  w[u]+=w[v];  
 }
 if(flag[u])w[u]++;
}
struct pi{
 int id;ll sum;
// bool operator <(const pi &t) const{
//  return sum<t.sum;
// }
}p[maxn];
int top=0;
bool cmp(pi a,pi b){
 return a.sum<b.sum;
}
inline void DFS(int u){
 ll l=top+1;
 for(int i=head[u];i;i=e[i].nxt){
  int v=e[i].v;
  p[++top]=(pi){v,w[v]}; 
 }
 ll r=top;
 ll all=r-l+1;
 if(all>0){
  sort(p+l,p+r+1,cmp);
  ans+=all;
  for(int i=l;i<=r;i++){
   ans+=1ll*(all-(i-l+1))*w[p[i].id];
   DFS(p[i].id);
  }
 }
}
char c[520000];
signed main(){
 ll n;
// scanf("%lld",&n);
 n=in;
 for(int i=1;i<=n;i++){
 
  scanf("%s",c);
  insert(c);
 }
 rego(0,0);
 DFS(0);
 printf("%lld",ans);
 
}