【Luogu P1364】 医院设置

题目描述

设有一棵二叉树，如图：

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

若医院建在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

=================================================================================================这次又是重构了一次代码就莫名过了qwq

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;

struct Node {
    int to,next;
}edge[2010];

int k[110],cnt,h[110],n,rem[110],sum,mn=99999999;
bool vis[110];

void add(int be,int en){
    if(be&&en){
        cnt++;
        edge[cnt].to=en;
        edge[cnt].next=h[be];
        h[be]=cnt;
    }
}

void dfs(int u,int ds){
    for(int i=h[u];i>0;i=edge[i].next){
        int v=edge[i].to;
        if(v!=0 && !vis[v]){
            vis[v]=1;
            rem[v]=ds+1;
            dfs(v,ds+1);
        }
    }
}

int main(){
    cin>>n;
    int be,en;
    for(int i=1;i<=n;i++){
        cin>>k[i]>>be>>en;
        add(i,be);add(be,i);
        add(i,en);add(en,i);
    }for(int i=1;i<=n;i++){
        memset(vis,0,sizeof vis);
        memset(rem,0,sizeof rem);
        sum=0;
        vis[i]=1;
        dfs(i,0);
        for(int j=1;j<=n;j++)
            sum+=k[j]*rem[j];
        if(sum<mn)    mn=sum;
    }    cout<<mn<<endl;
}