洛谷P1364医院设置

洛谷P1364 医院设置

医院设置

题目描述

设有一棵二叉树，如图：

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

5						
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0

样例输出 #1

81

提示

数据规模与约定

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

正确代码

//
// Created by 枫飘长安 on 2024/4/15.
//
#include <bits/stdc++.h>
using namespace std;
/**
  *author: 枫飘长安
  *date: 2024/4/15
  *fileName: P1364.cpp
  *description: TODO
  **/
typedef long long ll;
const int MAXN = 1e6+10;

int ans = 0x3f3f3f3f;
int n;
bool graph[105][105];
bool vis[105];
int num[105];


struct node {
    int u,step;
};

int bfs(int x) {
    memset(vis,0,sizeof(vis));
    queue<node> q;
    vis[x] = true;
    q.push((node){x,0});
    int sum = 0;
    while(!q.empty()){
        node temp=q.front();
        q.pop();
        for(int i=1;i<=n;i++)
            if(graph[temp.u][i]&&!vis[i]){
                vis[i] = true;
                q.push({i,temp.step+1});
                sum+=num[i]*(temp.step+1);
            }
    }
    return sum;
}

int main (){
    cin >> n;
    for (int i = 1; i <= n; i++) {
        int l,r;
        cin>> num[i] >> l >> r;
        if (l != 0) {
            graph[i][l] = true;
            graph[l][i] = true;
        }
        if (r != 0) {
            graph[i][r] = true;
            graph[r][i] = true;
        }
    }
    for (int i = 1; i <=n; i++) {
        ans = min(ans,bfs(i));
    }
    cout << ans;
    return 0;
}