最优布线问题(最小生成树)

最优布线问题

Description
学校有n台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。
当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。
现在由你负责连接这些计算机,你的任务是使任意两台计算机都连通(不管是直接的或间接的)。
Input
输入文件wire.in,第一行为整数n(2<=n<=100),表示计算机的数目。此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。
Output
输出文件wire.out,一个整数,表示最小的连接费用。
Sample Input
3
0 1 2
1 0 1
2 1 0
Sample Output
2(注:表示连接1和2,2和3,费用为2)
分析
这是一道最小生成树模板
可以用普里姆算法(prim)克鲁斯卡尔算法(kruskal)
因为这里的n<=100
所以我这里用的是prim算法
蓝,白点思想
AC代码

#include<iostream>
#include<cstring>
using namespace std;
int n,k,s,a[105][105],b[105],m[105];
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	 for(int j=1;j<=n;j++)
	  cin>>a[i][j];
	memset(m,0x7f,sizeof(m));//将值弄最大
	m[1]=0;  //将第一个为0
	for(int i=1;i<=n;i++)
	{
		k=0;
		for(int j=1;j<=n;j++)
		 if(!b[j]&&(m[j]<m[k]))//如果这是蓝点,并且与白点相连最短,就记录
		  k=j;
		b[k]=1;//将这个为白点
		for(int j=1;j<=n;j++)
		 if(!b[j]&&(a[k][j]<m[j]))//如果这是蓝点,并且与这个白点的距离,比与它相连的白点的值还小,就更新
		  m[j]=a[k][j];   
	}  
	for(int i=1;i<=n;i++)//加每个点的最小权值
	 s+=m[i];
	cout<<s;
} 

还有一个kruskal算法

#include<iostream>
using namespace std;
int n,m,x,y,s,o,v[105],a[105][105];
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	 for(int j=1;j<=n;j++)
	  cin>>a[i][j];
	for(int i=1;i<=n;i++)v[i]=i;  //给它们赋值初值
	for(int k=1;k<=n-1;k++)
  	 {
      	m=2147483647;//定大点,记住不能为
      	for(int i=1;i<=n;i++)//找两个点不在同一个集合中,最小的边
     	for(int j=1;j<=n;j++)
      	 if((v[i]!=v[j])&&(a[i][j]<m)&&(a[i][j]!=0))
	   	  {m=a[i][j];x=i;y=j;}
      	s+=m;
      	int t=v[y];//一定要这样
      	for(int i=1;i<=n;i++)
         if(v[i]==t)v[i]=v[x];//将这两个集合合并
	  }
  	cout<<s;
}

谢谢

发布了49 篇原创文章 · 获赞 79 · 访问量 1649

猜你喜欢

转载自blog.csdn.net/weixin_45524309/article/details/103639416