1349：【例4-10】最优布线问题

【题目描述】

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。

当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。

现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

【输入】

第一行为整数nn（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。

【分析】

样例图

 最小生成树

 

根据题目描述，可以用Prim算法，最小生成树生成后，把所有的边权加起来就是最终答案。

【代码实现】

#include<bits/stdc++.h>
using namespace std;
int n,sum; 
bool flag[109];
struct Node{
	int from,id,w;
	Node(){};
	Node(int argfrom,int argid,int argw){
		from=argfrom;
		id=argid;
		w=argw;
	}
	friend bool operator <(Node a,Node b){
		return a.w>b.w;
	}
};
priority_queue<Node>pque;
int g[109][109];
vector<Node>result;
void Prim(int sid){
	memset(flag,false,sizeof(flag));
	pque.push(Node(-1,sid,0));
	while(result.size()<n-1){
		Node nd=pque.top();
		pque.pop();
		if(flag[nd.id])continue;
		flag[nd.id]=true;
		if(nd.from!=-1)result.push_back(nd);
		for(int i=1;i<=n;i++){
			if(!flag[i]&&g[nd.id][i]<0x3f3f3f3f){
				pque.push(Node(nd.id,i,g[nd.id][i]));
			}
		}
	}
}
int main(){
	memset(g,0x3f,sizeof(g));
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>g[i][j];
		}
	}
	Prim(1);
	for(int i=0;i<result.size();i++){
		sum+=result[i].w;
	}
	cout<<sum;
	return 0; 
}