4567: [Scoi2016]背单词

4567: [Scoi2016]背单词

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 405   Solved: 147
[ Submit][ Status][ Discuss]

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

HINT

Source

[ Submit][ Status][ Discuss]



简明题意：现在有n个单词，要给它们确定一个排列顺序，使得学习的代价最小，代价统计如下

1.在第i个单词后面如果有一个单词是它的后缀，代价i*i

2.在第i个单词前后都没有它的后缀，代价i

3.在第i个单词后没有后缀，但前面有，设前面的后缀中位置最后面的一个为j，代价i - j

第一条肯定不能让他发生，因为1 + 2 + 3 + ... + n <= n*n

每个单词翻转，建一棵trie，删去没带表单词的无用节点，得到一棵树

对这棵树进行一次dfs，得到每个节点的dfs序，那么每个点产生的代价就是dfn[x] - dfn[fa[x]]

现在问题的目标是让所有代价和最小

这个统计是类似于差值的东西，不同子树内的dfs顺序不互相影响

可以发现，有影响的就是同一个节点的所有儿子的dfs顺序

假设对于节点x，第一个遍历的它的儿子是k，那么其余儿子的dfn至少要加上size[k]

因此先遍历k，对答案的影响，是增加了(son[x]-1)*size[k]

同理，如果第二个遍历t，对答案的影响，是增加了(son[x]-2)*size[t]....

以此下去，一个很明显的贪心，是每次选择size最小的子树先走过去，这样和一定最小

证明?排序不等式:逆序和<=乱序和<=顺序和

#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<bitset>
using namespace std;

const int maxn = 1E5 + 10;
const int maxm = 5E5 + 1E4 + 10;
typedef long long LL;

int n,m,cnt,rt,dfs_clock,dfn[maxn],siz[maxn],ch[maxm][26],Num[maxm];
LL Ans;
char s[maxm];

vector <int> v[maxn];

bool cmp(const int &x,const int &y) {return siz[x] < siz[y];}

void Build_tree(int x,int fa)
{
	if (Num[x]) v[fa].push_back(Num[x]),fa = Num[x];
	for (int i = 0; i < 26; i++)
		if (ch[x][i]) Build_tree(ch[x][i],fa);
}

void Dfs(int x)
{
	siz[x] = 1;
	for (int i = 0; i < v[x].size(); i++)
		Dfs(v[x][i]),siz[x] += siz[v[x][i]];
}

void Calc(int x)
{
	dfn[x] = ++dfs_clock;
	sort(v[x].begin(),v[x].end(),cmp);
	for (int i = 0; i < v[x].size(); i++)
	{
		int to = v[x][i];
		Calc(to); Ans += 1LL*(dfn[to] - dfn[x]);
	}
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		scanf("%s",s + 1); 
		int x = rt; m = strlen(s + 1); 
		for (int j = m; j; j--)
		{
			if (!ch[x][s[j]-'a']) ch[x][s[j]-'a'] = ++cnt;
			x = ch[x][s[j]-'a'];
		}
		Num[x] = i;
	}
	Build_tree(rt,rt); Dfs(rt); Calc(rt); cout << Ans;
	return 0;
}