[bzoj4567] [loj#2012] [SCOI2016] 背单词

Description

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

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

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

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

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

Input

输入一个整数，表示 $Lweb$ 要学习的单词数。接下来 $n$ 行，每行有一个单词（由小写字母构成，且保证任意单词两两互不相同）。

Output

$Lweb$ 吃的最少泡椒数。

Sample Input

2

a

ba

Sample Output

HINT

$1 \leq n \leq 10^5$，所有字符的长度总和 $1 \leq |len| \leq 510000$

想法

把所有串反过来，建在 $trie$ 树上，形成一颗前缀树，那么原题中的后缀即这里的前缀

一个显然的贪心是所有串都在它所有前缀填入后再填，$n \times n$ 的代价要不起。

相当于在树上进行遍历，使所有点的父节点都在子节点前访问，总代价是所有点与其父节点的访问时间差的总和。

于是又一个贪心，每个点访问后按它所有子节点 $size$ 的大小，从小到大依次访问。

很容易想到这个贪心，但咋证明是对的呢？【听说这是个经典问题，但网上也没看见多少证明啊（逃】

可以看 $neither_nor$ 的证明 https://blog.csdn.net/neither_nor/article/details/51362523

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 110000;
typedef long long ll;
int n;
int fa[N],sz[N];
struct trie{
	trie *ch[26];
	int id;
}pool[N*5],*root;
int cnt;
void insert(char s[],int id){
	trie *p=root;
	int len=strlen(s);
	for(int i=len-1;i>=0;i--){
		if(!p->ch[s[i]-'a']) p->ch[s[i]-'a']=&pool[++cnt];
		p=p->ch[s[i]-'a'];
	}
	p->id=id;
}
struct node{
	node *nxt;
	int v;
}pool2[N],*h[N];
int cnt2;
void addedge(int u,int v){
	node *p=&pool2[++cnt2];
	p->v=v;p->nxt=h[u];h[u]=p;
}
void dfs(trie *p,int cur){
	int nc=p->id ? p->id : cur;
	for(int i=0;i<26;i++)
		if(p->ch[i])
			dfs(p->ch[i],nc);
	if(p->id){
		sz[p->id]++;
		fa[p->id]=cur;
		sz[cur]+=sz[p->id];
		addedge(cur,p->id);
	}
}
ll ans;
int b[N],tot;
void work(int u){
	for(node *p=h[u];p;p=p->nxt) work(p->v);
	tot=0;
	for(node *p=h[u];p;p=p->nxt) b[++tot]=sz[p->v];
	sort(b+1,b+1+tot);
	for(int i=1;i<=tot;i++)
		ans+=1ll*b[i]*(tot-i);
	ans+=tot;
}
int main()
{
	char s[N];
	scanf("%d",&n);
	root=&pool[++cnt];
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		insert(s,i);
	}
	dfs(root,0);
	ans=0; work(0);
	printf("%lld\n",ans);
	return 0;
}