医院设置 codevs 2577

Floyd+二叉树

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
int a[101];
int g[101][101];
int main()
{
  int n,i,j,k,l,r,min,ans;
  cin>>n;
 for (i=1;i<=n;i++)
  for (j=1;j<=n;j++)
    g[i][j]=10000000;
  for (i=1;i<=n;i++)
  {
    g[i][i]=0;
    cin>>a[i]>>l>>r;
    if (l>0) g[i][l]=g[l][i]=1;
    if (r>0) g[i][r]=g[r][i]=1;
  }
  for (k=1;k<=n;k++)
    for (i=1;i<=n;i++)
      if (i!=k)
       for (j=1;j<=n;j++)
         if (i!=j&&k!=j&&g[i][k]+g[k][j]<g[i][j])
            g[i][j]=g[i][k]+g[k][j];
  min=0x7fffffff;
  for (i=1;i<=n;i++)
  {
    ans=0;
    for (j=1;j<=n;j++)
      ans+=g[i][j]*a[j];
    if (ans<min) min=ans;
  }
  cout<<min<<endl;
  return 0;
}