【bzoj 4567】[Scoi2016]背单词 trie树+贪心

因为要维护的是后缀，所以把字符串倒叙插入，然后其实第一种情况是一定不能发生的，所以在一个字符串插入之前必须将他的所有的后缀全部插入，然后证明两个单词之间是不会有影响的，交换任意两个没有后缀关系的单词，其中一个的贡献加一，而另外一个减一所以没有影响。所以建一个新树，再跑贪心就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#define maxn 700020
#define LL long long
using namespace std;
int n,son[maxn][26],tot,vis[maxn],size[maxn],now=-1;
LL ans;
vector<pair<int,int> >g[maxn];
char s[maxn];

void insert(){
int rt=0,len=strlen(s+1);
for(int i=len;i>=1;i--){
if(!son[rt][s[i]-'a'])son[rt][s[i]-'a']=++tot;
rt=son[rt][s[i]-'a'];
}
vis[rt]=1;
}

void dfs1(int u,int top=0){
int tt=top;
if(vis[u])size[u]=1,top=u;
for(int i=0;i<26;i++)if(son[u][i])dfs1(son[u][i],top),size[u]+=size[son[u][i]];
if(vis[u])g[tt].push_back(make_pair(size[u],u));
}
void dfs2(int u,int fa){
now++;
sort(g[u].begin(),g[u].end());
int p=now;
ans+=(LL)now-fa;
for(int v,i=0;i<g[u].size();i++){
v=g[u][i].second;
dfs2(v,p);
}
}

int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",s+1);
insert();
}
dfs1(0);
dfs2(0,0);
printf("%lld",ans);
return 0;
}