最优布线问题

题目描述

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

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

输入格式

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

输出格式

输出文件wire.out，一个整数，表示最小的连接费用。

输入样例 复制

3
0 1 2
1 0 1
2 1 0

输出样例 复制

2

数据范围与提示

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

方法一：

prime算法：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 110,null = 0x3f3f3f3f;

int n,res;
int g[N][N];
int dist[N];
bool st[N];

void prime()
{
	memset(dist,0x3f,sizeof dist);
	
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
			if(!st[j] && (t==-1 || dist[t]>dist[j]))
				t=j;
				
		if(i) res+=dist[t];
		
		for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
		
		st[t]=true;
	}
}

int main()
{
	cin >> n;
	memset(g,0x3f,sizeof g);
	
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			int a;
			cin >> a;
			if(a) g[i][j]=a;
		}
		
	prime();
	
	cout << res << endl;
	return 0;
}

方法二：

Kruskal算法：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 10010;

int n,m;
int p[N];

struct edge
{
	int a,b,w;
	
	bool operator< (const edge &W) const//将其先按w经行排序
	{
		return w<W.w;
	}
}edges[N];

int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}

int main()
{
	cin >> n;
	
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			int w;
			cin >> w;
			if(w) edges[m++]={i,j,w};
		}
		
	sort(edges,edges+m);
	
	for(int i=1;i<=n;i++) p[i]=i;
	
	int res=0;
	for(int i=0;i<m;i++)
	{
		int a=edges[i].a,b=edges[i].b,w=edges[i].w;
		a=find(a),b=find(b);
		if(a!=b)
		{
			p[a]=b;
			res+=w;
		}
	}
	cout << res << endl;
	return 0;
}