这道题快让我疯了,超过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;
}