题目意思有点难懂,给定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;
}