洛谷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;
}