[十二省联考2019] 春节十二响

Description

传送门

Solution

如果只有左右两条链，一定是左链和右链的最大配最大、次大配次大，两条链这样合并后就变成了一条链，然后继续按照这种方法跟其他的链匹配即可。

一开始是这么写的，但是迭代器 \(p\) 已经删掉了，再 --\(p\) 就会出奇怪的错误。

    p = s[a[u]].end(), q = s[a[e[i].v]].end(), --p, --q;
    for (;;) {
        if (*p < *q) s[a[u]].erase(p), s[a[u]].insert(*q);
        if (q == s[a[e[i].v]].begin()) break;
        --p, --q;
    }

Code

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

const int N = 200005;
struct Edge { int v, nxt; } e[N];
int f[N], m[N], head[N], a[N], tot, s[N], t;
std::priority_queue<int> q[N];
long long ans;

int read() {
	int x = 0; char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
	return x;
}
void adde(int u, int v) {
	e[++tot].nxt = head[u], head[u] = tot, e[tot].v = v;
}
void dfs(int u, int f) {
	a[u] = u;
	for (int i = head[u]; i; i = e[i].nxt) if (e[i].v != f) {
		dfs(e[i].v, u);
		if (q[a[e[i].v]].size() > q[a[u]].size()) std::swap(a[u], a[e[i].v]);
		while (!q[a[e[i].v]].empty())
			s[++t] = std::max(q[a[e[i].v]].top(), q[a[u]].top()), q[a[e[i].v]].pop(), q[a[u]].pop();
		while (t) q[a[u]].push(s[t]), --t;
	}
	q[a[u]].push(m[u]);
}
int main() {
	int n = read();
	for (int i = 1; i <= n; ++i) m[i] = read();
	for (int i = 2; i <= n; ++i) f[i] = read(), adde(f[i], i);
	dfs(1, 0);
	while (!q[a[1]].empty()) ans += q[a[1]].top(), q[a[1]].pop();
	printf("%lld\n", ans);
	return 0;
}