63 lines
2.0 KiB
C++
63 lines
2.0 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
const int N = 1e5 + 10, M = N << 1;
|
|||
|
|
const int INF = 0x3f3f3f3f;
|
|||
|
|
|
|||
|
|
// 链式前向星
|
|||
|
|
int e[M], h[N], idx, w[M], ne[M];
|
|||
|
|
void add(int a, int b, int c = 0) {
|
|||
|
|
e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
|
|||
|
|
}
|
|||
|
|
int c[N];
|
|||
|
|
int f[N], sz[N];
|
|||
|
|
int ans = INF;
|
|||
|
|
|
|||
|
|
// 第一次dfs,获取在以1为根的树中:
|
|||
|
|
// 1、每个节点分别有多少个子节点，填充sz[]数组
|
|||
|
|
// 2、获取到f[1],f[1]表示在1点设置医院的代价
|
|||
|
|
// 获取到上面这一组+一个数据，才能进行dfs2进行换根
|
|||
|
|
void dfs1(int u, int fa, int step) {
|
|||
|
|
sz[u] = c[u]; // 这个挺绝啊~，与一般的统计子树节点个数不同，这里把人数，也就是点权值，也看做是一个节子点，想想也是这个道理
|
|||
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|||
|
|
int v = e[i];
|
|||
|
|
if (v == fa) continue;
|
|||
|
|
dfs1(v, u, step + 1); // 填充深搜v节点为根的子树
|
|||
|
|
sz[u] += sz[v]; // 在完成了v节点的数据统计后，用v节点的sz[v]结果累加到sz[u]
|
|||
|
|
}
|
|||
|
|
f[1] += step * c[u]; // 累加步数*人数 = 1点的总代价,预处理出1点的总代价
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 第二次dfs,开始dp换根
|
|||
|
|
void dfs2(int u, int fa) {
|
|||
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|||
|
|
int v = e[i];
|
|||
|
|
if (v == fa) continue;
|
|||
|
|
f[v] = f[u] + sz[1] - sz[v] * 2; // 经典的递推式
|
|||
|
|
dfs2(v, u); // 继续深搜
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
// 初始化链式前向星
|
|||
|
|
memset(h, -1, sizeof h);
|
|||
|
|
|
|||
|
|
int n;
|
|||
|
|
cin >> n;
|
|||
|
|
for (int i = 1; i <= n; i++) {
|
|||
|
|
cin >> c[i];
|
|||
|
|
int a, b;
|
|||
|
|
cin >> a >> b;
|
|||
|
|
if (a) add(a, i), add(i, a); // 是一个二叉树结构，与左右节点相链接，但有可能不存在左或右节点，不存在时,a或b为0
|
|||
|
|
if (b) add(b, i), add(i, b);
|
|||
|
|
}
|
|||
|
|
// 1、准备动作
|
|||
|
|
dfs1(1, 0, 0);
|
|||
|
|
// 2、换根dp
|
|||
|
|
dfs2(1, 0);
|
|||
|
|
|
|||
|
|
// 输出答案
|
|||
|
|
for (int i = 1; i <= n; i++) ans = min(ans, f[i]);
|
|||
|
|
cout << ans << endl;
|
|||
|
|
return 0;
|
|||
|
|
}
|