【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;
}