信息学奥赛一本通1349-最优布线问题

【题目描述】

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

---

【输入】

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

【输出】

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

---

【输入样例】

3
0 1 2
1 0 1
2 1 0

【输出样例】

2

---

【提示】

注：表示连接1和2，2和3，费用为2。

---

解题思路

就是最小生成树而已，这里用时间复杂度是O(n^2)的算法（既Prim法）。

---

代码

#include<cstdio>
using namespace std;
int n,k,cost[101][101],lowcost[101],x,s;
bool ok[101];
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
      for (int j=1;j<=n;j++) cost[i][j]=23333333;
    //初始化无穷大
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=n;j++)
        {
            scanf("%d",&cost[i][j]);
            //输入权值
        }
    }
    for (int i=1;i<=n;i++) lowcost[i]=cost[1][i];
    //记录每个点到目前集合最小花费
    ok[1]=true;//标记为以搜过
    for (int i=2;i<=n;i++)
    {
        int k=0,mins=23333333;
        for (int j=1;j<=n;j++)
          if (!ok[j] && lowcost[j]<mins)
          {
            mins=lowcost[j];
            k=j;//找并且记录最小值
          }
        ok[k]=true;//标记，更新集合
        s+=lowcost[k];//记录总值
        for (int j=1;j<=n;j++)
          if (lowcost[j]>cost[k][j]) lowcost[j]=cost[k][j]; 
        //更新每个点到目前集合最小花费
    }
    printf("%d",s);
}