nyoj--38(布线问题)

这道题快让我疯了,超过8次Wronganswer大哭,后来终于发现是自己对于Prim算法的理解不够全面,这道题中存在几个楼是未连接的,所以在赋初值时,这些未连接的楼之间应该设费用为无穷大,否则对后来判断最优值(费用最小值)有影响。

还有在对邻接矩阵赋初值时,可以巧用0x3f3f3f3f,即memset(mark,0x3f,sizeof(mark));

布线问题

时间限制: 1000 ms  |  内存限制: 65535 KB
难度: 4

描述 南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
1、把所有的楼都供上电。
2、所用电线花费最少

输入 第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里,每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。(哪两栋楼间如果没有指明花费,则表示这两栋楼直接连通需要费用太大或者不可能连通)
随后的1行里,有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0<e<v*(v-1)/2 )
(楼的编号从1开始),由于安全问题,只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。 输出 每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。 样例输入

1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6

样例输出:

4

代码如下:Prim算法

#include<stdio.h>
#include<string.h>
#define M 0x3f3f3f
int mark[510][510];
int vist[510];
int low[510];
int v,e;
int prim()
{
	int i,j,temp,k,sum;
	sum=0;
	vist[1]=1;
	for(i=1;i<=v;i++)
	{
		low[i]=mark[1][i];
	}
	for(i=1;i<=v;i++)
	{
		temp=M+10;
		for(j=1;j<=v;j++)
		{
			if(!vist[j]&&temp>low[j])
			{
				k=j;
				temp=low[j];
			}
		}
		if(temp==M+10)break;
		sum+=low[k];
		vist[k]=1;
		for(j=1;j<=v;j++)
		{
			if(!vist[j]&&low[j]>mark[k][j])
			low[j]=mark[k][j];
		}
	}
	return sum;
}
int main()
{
	int t,x,y,z,m,min,i,j,toal;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&v,&e);
		memset(mark,0x3f,sizeof(mark));
		for(i=1;i<=e;i++)
		{
			scanf("%d%d%d",&x,&y,&z);
			mark[x][y]=z;
			mark[y][x]=z;
		}
		min=M;
		for(i=1;i<=v;i++)
		{
			scanf("%d",&m);
			if(min>m)
			min=m;
		}
		memset(vist,0,sizeof(vist));
		 toal=prim();
		printf("%d\n",toal+min);
	}
	return 0;
}

Kruscal算法:

#include<stdio.h>
#include<algorithm>
using namespace std;
struct stu
{
	int s;
	int u;
	int w;
}vt[130000];
int per[510];
int cross[510];
int cmp1(stu a,stu b)
{
	return a.w<b.w;
}
int cmp2(int a,int b)
{
	return a<b;
}
int find(int x)
{
	return x==per[x]?x:per[x]=find(per[x]);
}
bool join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		per[fx]=fy;
		return true;
	}
	else return false;
}
int main()
{
	int t,i,v,e;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&v,&e);
		for(i=0;i<e;i++)
			scanf("%d%d%d",&vt[i].s,&vt[i].u,&vt[i].w);
			  for(i=0;i<v;i++)
			  {
			  scanf("%d",&cross[i]);
			  per[i]=i;
		      }
		      per[i]=i;
		      int sum=0;
		      int flag=0;
		      sort(vt,vt+e,cmp1);
		      for(i=0;i<e;i++)
		      {
		      	if(join(vt[i].s,vt[i].u))
		      	{
		      		flag++;
		      		sum+=vt[i].w;
		      	}
		      	if(flag==v-1)break;
		      }
		      sort(cross,cross+v,cmp2);
			  sum+=cross[0];
			  printf("%d\n",sum);
	}
	return 0;
}



 

你可能感兴趣的:(nyoj--38(布线问题))