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
,top;ll Ans,ord
,sum;
int son[M][26],f[M],_cnt=1;//tire
int size
,first
,next
,to
,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;
}