首页 > 代码库 > luogu P1364 医院设置

luogu P1364 医院设置

P1364 医院设置

题目描述

设有一棵二叉树,如图:

技术分享                                         

其中,圈中的数字表示结点中居民的人口。圈边上数字表示结点编号,现在要求在某个结点上建立一个医院,使所有居民所走的路程之和为最小,同时约定,相邻接点之间的距离为l。如上图中,

若医院建在1 处,则距离和=4+12+2*20+2*40=136;若医院建在3 处,则距离和=4*2+13+20+40=81……

输入输出格式

输入格式:

第一行一个整数n,表示树的结点数。(n≤100)

接下来的n行每行描述了一个结点的状况,包含三个整数,整数之间用空格(一个或多个)分隔,其中:第一个数为居民人口数;第二个数为左链接,为0表示无链接;第三个数为右链接。

输出格式:

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

输入输出样例

输入样例#1:
5						
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0
输出样例#1:
81

化树为图:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;
const int N=200;
const int Maxn=9999999;

int now=1;
int head[N];
int dis[N];
int people[N];
bool vis[N];
int n,x,y;
queue<int>q;

struct node{
    int u,v,w,nxt;
}E[N];

inline int read()
{
    int x=0;
    char c=getchar();
    while(c<0||c>9)c=getchar();
    while(c>=0&&c<=9)x=x*10+c-0,c=getchar();
    return x;
}

inline void add(int u,int v)
{
    E[now].v=v;
    E[now].w=1;
    E[now].nxt=head[u];
    head[u]=now++;
}

inline void spfa(int start)
{
    for(int i=1;i<=n;i++)
        dis[i]=Maxn,vis[i]=0;
    dis[start]=0;
    vis[start]=1;
    q.push(start);
    while(!q.empty())
    {
        int top=q.front();
        q.pop();
        vis[top]=0;
        for(int i=head[top];i!=-1;i=E[i].nxt)
        {
            if(dis[E[i].v]>dis[top]+E[i].w)
            {
                dis[E[i].v]=dis[top]+E[i].w;
                if(!vis[E[i].v])
                    vis[E[i].v]=1,q.push(E[i].v);
            }
        }
    }    
}

int main()
{
    n=read();
    for(int i=1;i<=n;i++)
        head[i]=-1;
        
    for(int i=1;i<=n;i++)
    {
        people[i]=read();
        x=read();
        y=read();
        if(x)
            add(i,x),add(x,i);
        if(y)
            add(i,y),add(y,i);
    }
    int answer=Maxn;
    for(int i=1;i<=n;i++)
    {
        spfa(i);
        int ans=0;
        for(int j=1;j<=n;j++)
            ans+=dis[j]*people[j];
        answer=min(ans,answer);
    }    
    printf("%d",answer);
    return 0;
}

 

luogu P1364 医院设置