摘果果

摘果果

题意

给出一棵以 \(1\) 为根的树和两个序列 \(a\) 和 \(b\)。

确定一种 DFS 遍历顺序，使得用以下方法计算出的权值最大：

1. 初始时 \(v \leftarrow 0,k\leftarrow 0\)。

2. 经过一个节点 \(x\) 时 \(v\leftarrow v+ka_x,k\leftarrow k+b_x\)。

3. 遍历完后最终的 \(v\) 即权值。

思路

考虑将每个节点的儿子排序。

先求出 \(A_x,B_x\) 表示 \(x\) 子树内 \(a\) 和 \(b\) 的和。

若两个儿子 \(x,y\) 满足 \(A_y \times B_x > A_x \times B_y\) 则 \(x\) 应在 \(y\) 前遍历。

左边表示先遍历 \(x\) 对 \(y\) 造成的贡献，右边表示先遍历 \(y\) 对 \(x\) 造成的贡献。

排序计算贡献即可。

代码

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 5;
struct node {
	int a, b;
};
int n, ans, A[N], B[N], nowb;
node p[N];
vector <int> E[N];
bool cmp(int x, int y) {
	return p[x].a * p[y].b < p[y].a * p[x].b;
}
void dfs1(int x, int fa) {
	for (auto y : E[x]) {
		if (y == fa) continue;
		dfs1(y, x);
		p[x].a += p[y].a;
		p[x].b += p[y].b;
	}
	sort(E[x].begin(), E[x].end(), cmp);
} 
void dfs2(int x, int fa) {
	ans += A[x] * nowb;
	nowb += B[x];
	for (auto y : E[x]) {
		if (y == fa) continue;
		dfs2(y, x);
	}
}
void solve() {
	cin >> n;
	for (int i = 1; i <= n; i ++) cin >> p[i].a, A[i] = p[i].a;
	for (int i = 1; i <= n; i ++) cin >> p[i].b, B[i] = p[i].b;
	for (int i = 1; i < n; i ++) {
		int u, v;
		cin >> u >> v;
		E[u].push_back(v);
		E[v].push_back(u);
	}
	dfs1(1, 0);
	dfs2(1, 0);
	cout << ans << "\n";
}
signed main() {
	int Case = 1;
//	cin >> Case;
	while (Case --)
		solve();
	return 0;
}