洛谷 P1364 医院设置题解带权树的重心

最新推荐文章于 2024-07-04 11:41:09 发布

bbc_ouo

最新推荐文章于 2024-07-04 11:41:09 发布

阅读量1.2k

点赞数 34

分类专栏：洛谷题解图论文章标签： c++ 算法图论深度优先

本文链接：https://blog.csdn.net/bbc_plus/article/details/138231159

版权

洛谷题解同时被 2 个专栏收录

51 篇文章 1 订阅

订阅专栏

图论

23 篇文章 0 订阅

订阅专栏

医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为 $1$ 。如上图中，若医院建在 $1$ 处，则距离和 $=4+12+2\times20+2\times40=136$ ；若医院建在 $3$ 处，则距离和 $=4\times2+13+20+40=81$ 。

输入格式

第一行一个整数 $n$ ，表示树的结点数。

接下来的 $n$ 行每行描述了一个结点的状况，包含三个整数 $w, u, v$ ，其中 $w$ 为居民人口数， $u$ 为左链接（为 $0$ 表示无链接）， $v$ 为右链接（为 $0$ 表示无链接）。

输出格式

一个整数，表示最小距离和。

样例 #1

样例输入 #1

样例输出 #1

数据规模与约定

对于 $100\%$ 的数据，保证 $\leq n \leq 100$ ， $\leq u, v \leq n$ ， $\leq w \leq 10^5$ 。

原题

洛谷P1364——传送门

思路

具体思路详见Huami360大佬的题解洛谷题解——传送门。
本蒟蒻借鉴大佬的思路，写了可读性较高的代码，希望能帮助到读者。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn = 1e2 + 6;
const int maxm = 5e5 + 6;

struct edge
{
    int to; // to为边的指向
};

vector<edge> e[maxn]; // 存储以点i为起点的边

int n;
int sum[maxn];  // 以u为根的子树的带权结点数
bool vis[maxn]; // 是否已访问过该结点
int w[maxn];    // 每个点的权值
int ans[maxn];  // 在第i个结点建立医院后的路程和

void dfs(int u, int depth) // 计算ans[1]，计算所有结点的sum值
{
    int cur_sum = w[u]; // 以u为根的子树的带权结点数，初始化为该结点的权值
    vis[u] = true;      // 记录已访问

    for (int i = 0; i < e[u].size(); i++) // 遍历以u为起点的所有边
    {
        int v = e[u][i].to; // 边的终点
        if (vis[v])
            continue;
        dfs(v, depth + 1); // 递归下一层，深度(也就是到医院的距离)要加1
        cur_sum += sum[v];
    }

    sum[u] = cur_sum;

    ans[1] += w[u] * depth; // 统计路程和，每个结点要加上 居民人口乘以到医院的距离
}

void dp(int u)
{
    vis[u] = true; // 记录已访问

    for (int i = 0; i < e[u].size(); i++) // 遍历以u为起点的所有边
    {
        int v = e[u][i].to; // 边的终点
        if (vis[v])
            continue;
        ans[v] = ans[u] + sum[1] - sum[v] - sum[v]; // 转移方程
        dp(v);
    }
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    int a, b;
    for (int i = 1; i <= n; i++)
    {
        cin >> w[i] >> a >> b;
        // 若非0，则用邻接表存储该无向边所对应的两条有向边
        if (a)
        {
            e[i].push_back({a});
            e[a].push_back({i});
        }
        if (b)
        {
            e[i].push_back({b});
            e[b].push_back({i});
        }
    }
    dfs(1, 0);
    memset(vis, 0, sizeof(vis)); // 记得重置vis数组
    dp(1);
    int min_ans = INT_MAX; // 最小距离和，初始化为最大
    for (int i = 1; i <= n; i++)
    {
        min_ans = min(min_ans, ans[i]); // 更新min_ans为所有距离和中最小的
    }
    cout << min_ans << '\n';

    return 0;
}