AT_abc409_e [ABC409E] Pair Annihilation
题目描述
[problemUrl]: https://atcoder.jp/contests/abc409/tasks/abc409_e
给定一棵包含 $N$ 个顶点的树。顶点编号为 $1,2,\dots,N$,边编号为 $1,2,\dots,N-1$。边 $j$ 双向连接顶点 $u_j$ 和 $v_j$,其权重为 $w_j$。此外,顶点 $i$ 上有一个整数 $x_i$:若 $x_i > 0$,表示该顶点有 $x_i$ 个正电子;若 $x_i < 0$,表示有 $-x_i$ 个电子;若 $x_i = 0$,则该顶点没有粒子。题目保证 $\sum_{i=1}^N x_i = 0$。
每次可以沿边 $j$ 移动 1 个正电子或电子,消耗能量 $w_j$。当正电子和电子位于同一顶点时,它们会相互抵消(数量相等时完全消失)。
求使所有正电子和电子完全消失所需的最小总能量。
输入格式
输入通过标准输入给出,格式如下:
> $N$
> $x_1$ $x_2$ $\dots$ $x_N$
> $u_1$ $v_1$ $w_1$
> $u_2$ $v_2$ $w_2$
> $\vdots$
> $u_{N-1}$ $v_{N-1}$ $w_{N-1}$
输出格式
输出答案。
说明/提示
### 约束条件
- $2 \leq N \leq 10^5$
- $|x_i| \leq 10^4$
- $\sum_{i=1}^N x_i = 0$
- $1 \leq u_j < v_j \leq N$
- $0 \leq w_j \leq 10^4$
- 给定的图是一棵树
- 输入均为整数
### 样例解释 1
初始状态 $x=(-3,+2,+2,-1)$。通过以下操作可以用最小能量 $9$ 使所有粒子消失:
1. 将顶点 $1$ 的 1 个电子移动到顶点 $2$,消耗能量 $2$,$x=(-2,+1,+2,-1)$;
2. 将顶点 $2$ 的 1 个正电子移动到顶点 $1$,消耗能量 $2$,$x=(-1,0,+2,-1)$;
3. 将顶点 $4$ 的 1 个电子移动到顶点 $1$,消耗能量 $3$,$x=(-2,0,+2,0)$;
4. 将顶点 $1$ 的 1 个电子移动到顶点 $3$,消耗能量 $1$,$x=(-1,0,+1,0)$;
5. 将顶点 $1$ 的 1 个电子移动到顶点 $3$,消耗能量 $1$,$x=(0,0,0,0)$。
无法用 $8$ 或更少能量完成目标,因此答案为 $9$。
### 样例解释 2
初始状态已满足条件时,输出 $0$。
翻译由 DeepSeek V3 完成