标签:ext name memset ems set 包含 之间 int sam
设有一棵二叉树,如图:
其中,圈中的数字表示结点中居民的人口。圈边上数字表示结点编号,现在要求在某个结点上建立一个医院,使所有居民所走的路程之和为最小,同时约定,相邻接点之间的距离为1。如上图中,
若医院建在1 处,则距离和=4+12+2*20+2*40=136;若医院建在3 处,则距离和=4*2+13+20+40=81……
第一行一个整数n,表示树的结点数。(n≤100)
接下来的n行每行描述了一个结点的状况,包含三个整数,整数之间用空格(一个或多个)分隔,其中:第一个数为居民人口数;第二个数为左链接,为0表示无链接;第三个数为右链接。
输出格式:一个整数,表示最小距离和。
5 13 2 3 4 0 0 12 4 5 20 0 0 40 0 0
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; }
标签:ext name memset ems set 包含 之间 int sam
原文地址:http://www.cnblogs.com/labelray/p/7953532.html