摘果果

摘果果

题意

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

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

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

2. 经过一个节点 $x$ 时 $v\leftarrow v+k{a}_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;
}