【题目描述】
学校有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);
}