1. 程式人生 > 其它 >【SSL】1614醫院設定

【SSL】1614醫院設定

技術標籤:圖論

【SSL】1614醫院設定

Time Limit:1000MS
Memory Limit:65536K

Description

設有一棵二叉樹(如右圖)。其中,圈中的數字表示結點中居民的人口。圈邊上數字表示結點編號,現在要求在某個結點上建立一個醫院,使所有居民所走的路程之和為最小,同時約定,相鄰接點之間的距離為1。如 右圖中,若醫院建在:
  1處,則距離和=4+12+220+240=136
  3處,則距離和=4*2+13+20+40=81
    ………….
在這裡插入圖片描述

Input

第一行一個整數n,表示樹的結點數。(n<=100)
接下來的n行每行描述了一個結點的狀況,包含三個整數,整數之間用空格(一個或多個)分隔,其中:第一個數為居民人口數;第二個數為左連結,為0表示無連結;第三個數為右連結。

Output

一個整數,表示最小距離和。

Sample Input

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

Sample Output

81

思路

先建好邊。
Floyd做最短路。

程式碼

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<cmath>
#include<queue>
#include<cstring>
using namespace std;
long long n,
b[110]; long long a[2010][2010]; void input()//建邊 { long long i,j,l,r; scanf("%lld",&n); for(i=1;i<=n;i++) for(j=1;j<=n;j++) a[i][j]=100000000; for(i=1;i<=n;i++) { scanf("%lld%lld%lld",&b[i],&l,&r); if(l) a[i][l]=a[l][i]=1; if(r) a[i][r]=a[r][i]=1;
} for(i=1;i<=n;i++) a[i][i]=0; return; } void floyd()//求多源最短路 { int i,j,k; for(k=1;k<=n;k++) for(i=1;i<=n;i++) for(j=1;j<=n;j++) if(a[i][k]+a[k][j]<a[i][j]) a[i][j]=a[i][k]+a[k][j]; return; } int main() { long long i,j,sum,ans=100000000; input(); floyd(); for(i=1;i<=n;i++) { for(sum=0,j=1;j<=n;j++)//計算距離和 sum+=b[j]*a[i][j]; ans=min(ans,sum); } printf("%lld",ans); return 0; }