P1364 医院设置
https://www.luogu.com.cn/problem/P1364
题意
- 一个树形村庄。一个节点内数字代表,村庄的人数。
- 求把医院放到哪里。大家所走的距离最小
想法
- BFS + 邻接表
- 暴力枚举每一个点做为医院的,最小路径
// P1364 医院设置
// Created by majoe on 2020/5/31.
// https://www.luogu.com.cn/problem/P1364
#include <bits/stdc++.h>
using namespace std;
const int N = 2e2 +10;
int n,vis[N];
int ne[N],e[N],h[N],w[N],idx;
void add(int a,int b){ // a到b有边
e[idx] = b;
ne[idx] = h[a];
h[a] = idx++;
}
struct node{
int num, steps;
};
int bfs(int x){ //bfs找当前点x为医院设置点时的总距离
memset(vis,0,sizeof(vis));
queue<node> q;
vis[x]=1;
q.push((node){x,0});
int sum=0;
while(!q.empty()){
node now=q.front();
q.pop();
for (int i = h[now.num]; i != -1; i = ne[i]) {
int t = e[i];
if(!vis[t]){ // 如果t没被访问过
sum += w[t] * (now.steps + 1);
vis[t] = 1;
q.push((node){t,now.steps+1});
}
}
}
return sum;
}
int main(){
cin >> n;
memset(h,-1,sizeof(h));
for (int i = 1; i <= n; ++i) {
int v , a , b;
cin >> v >> a >> b;
w[i] = v; //某点的人数
if(a){
add(i,a); add(a,i);
}
if(b) {
add(i,b); add(b,i);
}
}
int ans = 1e9;
for (int i = 1; i <= n ; ++i) {
ans = min(ans,bfs(i));
}
cout << ans;
return 0;
}