bzoj4567: [Scoi2016]背单词

题目意思有点难懂，给定n个单词，找到一种排列使代价最小。在x处的一个单词的代价为：1.有一个单词是它的后缀且在它后面n*n，2. 没有单词是它的后缀且在它后面x-前一个是它的后缀的单词的位置（如没有即为零）。

很显然可以倒着暴力构造一棵tire，由于1的代价太大，考虑不存在1。即每个后缀填完后才能填自己，同时填完一个单词的子树的最小代价是固定的，只要合理安排儿子的遍历顺序即可。显然应dfs先遍历size较小的儿子，总代价小于1+2+...+n<n*n。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define N 100005
#define M 600005
using namespace std;
int n,q[N],top;ll Ans,ord[N],sum;
int son[M][26],f[M],_cnt=1;//tire
int size[N],first[N],next[N],to[N],l,cnt=1;
char s[M];
bool cmp(int x,int y){return size[x]<size[y];}
void link(int x,int y)
{
	to[++l]=y;next[l]=first[x];first[x]=l;
}
void ins(char s[])
{
	int l=strlen(s),k=1;
	for (int i=l-1;i>=0;i--)
		if (son[k][s[i]-'a']) k=son[k][s[i]-'a'];
		else k=son[k][s[i]-'a']=++_cnt;
	f[k]=1;
}
void build(int k,int x)
{
	if (f[x]) link(k,++cnt),size[k=cnt]=1;
	for (int i=0;i<26;i++)
		if (son[x][i]) build(k,son[x][i]);
}
void Get(int x)
{
	for (int i=first[x];i;i=next[i])
		Get(to[i]),size[x]+=size[to[i]];
}
void dfs(int k,int w)
{
	sum++;Ans+=sum-w;w=sum;
	int l=top+1,r=top;
	for (int i=first[k];i;i=next[i])
		q[++r]=to[i];
	sort(q+l,q+r+1,cmp);
	top=r;
	for (int i=l;i<=r;i++)
		dfs(q[i],w);
	top=l-1;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
		scanf("%s",s),ins(s);
	build(1,1);Get(1);dfs(1,1);
	printf("%lld\n",Ans);
	return 0;
}