P1364 医院设置 题解

医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为 $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

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

样例输出 #1

81

提示

数据规模与约定

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

---

思路

数据量较小，直接对每个点bfs搜索其父节点和子节点即可。

#include<bits/stdc++.h>
using namespace std;
int n;
struct Node{
    int value,left,right,father,len;
}t[1010];
void build(int x,int v,int l,int r){
    t[x].value=v;
    t[x].left=l;t[x].right=r;
    t[l].father=x;
    t[r].father=x;
}
long long bfs(int x){
    queue<int> q;
    bool gone[1010];
    memset(gone,0,sizeof(gone));
    gone[x]=1;
    q.push(x);
    long long ans=0;
    t[x].len=0;
    while(!q.empty()){
        int x_now=q.front();
        ans+=t[x_now].value*t[x_now].len;
        q.pop();
        int x_nex[3]={t[x_now].father,t[x_now].left,t[x_now].right};
        for(int i=0;i<3;i++){
            int x0=x_nex[i];
            if(x0!=0&&!gone[x0]){
                q.push(x0);
                gone[x0]=1;
                t[x0].len=t[x_now].len+1;
            }
        }
    }
    return ans;
}
int main(){
    cin>>n;
    t[1].father=0;
    for(int i=1;i<=n;i++){
        int w,u,v;
        cin>>w>>u>>v;
        build(i,w,u,v);
    }
    long long ans=bfs(1);
    for(int i=2;i<=n;i++){
        ans=min(ans,bfs(i));
    }
    cout<<ans;
    return 0;
}