最优布线问题

题目

题目描述

   学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。  
   当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。
  现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

输入

第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

输出

一个整数，表示最小的连接费用。

输入样例 复制

3
0 1 2
1 0 1
2 1 0

输出样例 复制

2

说明

 

分析

   最小生成树模板

 

代码

   prime

#include<bits/stdc++.h>
using namespace std;
int a[101][101],wz;
bool pd[101];
int dis[101],minn;
int n,x,ans=0;
void fun()
{
    cin>>n;
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      cin>>a[i][j];
      
      memset(dis,0x1f,sizeof(dis));
      dis[1]=0;
      for (int i=1;i<=n;i++)   //每次找到最小的一个点
      {
           minn=1e9;
           for (int j=1;j<=n;j++)
           {
               if (pd[j]==0&&dis[j]<minn)
               {
                   minn=dis[j];
                   wz=j;
               }
           }
           pd[wz]=1;
          ans+=minn;
             for (int j=1;j<=n;j++)  //将最小的点与之相连的边跟新值
               if (pd[j]==0&&a[wz][j]>0)
                    dis[j]=min(dis[j],a[wz][j]);
           
      }
      
}
int main()
{
    fun();
    cout<<ans;
}