loj#2012. 「SCOI2016」背单词

题目链接

loj#2012. 「SCOI2016」背单词

题解

题面描述有点不清楚.

考虑贪心

type1的花费一定不会是优的，不考虑，

所以先把后缀填进去,对于反串建trie树，

先填父亲再填儿子,这样每个单词的后缀填完了才会被填。

不是单词结束点的点是没用的，去掉

根据直觉，填单词和dfs序有关，所以应该先填Size小的

根据贪心，先填Size小的儿子。因为将Size小的先填可以减少后面儿子的代价

而先填大的会增加代价。

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm> 

inline int read() {
	int x = 0,f = 1;
	char c = getchar();
	while(c < '0' || c > '9')c = getchar();
	while(c <= '9' && c >= '0') x = x * 10 + c - '0',c = getchar();
	return x * f;
}
const int maxn = 500007;
char s[maxn];
int tag[maxn],ch[maxn][26];
int tot = 0;
void insert(char *s) {
	int len = strlen(s + 1),now = 0;
	for(int i = len;i >= 1;-- i) {
		int t = s[i] - 'a';
		if(!ch[now][t]) ch[now][t] = ++ tot;
		now = ch[now][t];
	}
	tag[now] = 1;
} 

struct node {
	int v,next;
} edge[maxn << 1];
int num = 0,head[maxn];
inline void add_edge(int u,int v) {
	edge[++ num].v = v;edge[num].next = head[u],head[u] = num;
}
int siz[maxn];
int size[maxn];
void dfs(int x,int fa) {
	if(tag[x]) siz[fa] ++,add_edge(fa,x),fa = x;
	for(int i = 0;i < 26;++ i) if(ch[x][i]) dfs(ch[x][i],fa);
	size[x] = siz[x];
	for(int i = head[x];i;i = edge[i].next) {
		size[x] += size[edge[i].v];
	}
}
#define pr std::pair<int,int>
#define mp std::make_pair
long long ans = 0;
std:: priority_queue<pr>q;
void solve() {
	q.push(mp(0,0)); int cnt = 1;
	while(!q.empty()) {
		int x = q.top().second; q.pop();
		cnt = cnt + siz[x] - 1;
		ans += cnt;
		for(int i = head[x];i;i = edge[i].next) {
			q.push(mp(-size[edge[i].v],edge[i].v));
		}
	}
}
int main() {
	int n  = read();
	for(int i = 1;i <= n;++ i) {
		scanf("%s",s + 1);
		insert(s);
	}
	dfs(0,0);
	solve();
	printf("%lld\n",ans);
	return 0;
}