P1364 医院设置 - bfs

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