【prim算法】【最小生成树】【图论】最优布线问题

最优布线问题

题目

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

输入

输入第一行为整数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）

解题思路

其实这道题目就是用prim算法去不断的进行，这个算发和Dijkstra算法很像，但赋的值不一样

程序如下

$$邻接矩阵$$

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
int n,a[1001][1001],f[1001],t,ans;
bool b[1001];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=n;++j)
		{
			scanf("%d",&a[i][j]);
		}
	}
	memset(b,1,sizeof(b));
	memset(f,0x7f,sizeof(f));
	f[1]=0;
	for(int i=1;i<=n;++i)
	{
		t=0;
		for(int j=1;j<=n;++j)
		{
			if(b[j]&&f[j]<f[t])//找一个与白点的相连并最小的
			{
				t=j;
			}
		}
		b[t]=false;
		for(int j=1;j<=n;++j)
		{
			if(b[j]&&a[t][j]<f[j])//把蓝点加入生成树
			{
				f[j]=a[t][j];
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
		ans+=f[i];//权值累加
	}
	printf("%d",ans);
	return 0;
}

$$邻接表$$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,x,s,a[1000005],b[1000005],r[1000005],t,ans;
struct q
{
	int l,to,next;
}f[1000005];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=n;++j)
		{
			scanf("%d",&x);
			if (!x) continue;
			f[++s]=(q){x,j,a[i]};
			a[i]=s;
		}
	}
	memset(r,0x7f,sizeof(r));
	r[1]=0;
	for(int i=1;i<=n;++i)
	{
		t=0;
		for(int j=1;j<=n;++j)
		{
			if(!b[j]&&r[j]<r[t]) t=j;//同上
		}
		b[t]=true;
		ans+=r[t];
		for(int j=a[t];j;j=f[j].next)
		{
			if(f[j].l<r[f[j].to])
				r[f[j].to]=f[j].l;//同上
		}
	}
	printf("%d",ans);
	return 0;
}