HDU-1879(prim构最小生成树)

这个题目是一个很基础的求最小生成树的,唯一稍微拐一点弯的就是,你个建成边的处理情况,可能有很多种情况,但我选择的是:既然已经建成了,那么说明他的花费是0,直接以0为权值,建到矩阵中就可以,而且这样建的话也不会更新这些已经建成的边,原因很简单咯,因为你最初设这的dis[i]=0;

贴上代码:

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#define inf 0x3fffffff

int N,M;//N represents the number of points,and M represents the number of edges;
int visit[105];
int map[105][105];
int  dis[105];

void prim()
{
	for(int i=1;i<=N;i++)
		dis[i]=inf;
	dis[1]=0;
	for(int j=1;j<=N;j++)
	{
		int t=inf,pos;
		for(i=1;i<=N;i++)
		{
			if(!visit[i]&&t>dis[i])
			{
				t=dis[i];
				pos=i;
			}
		}
		visit[pos]=1;
		for(i=1;i<=N;i++)
		{
			if(!visit[i]&&dis[i]>map[pos][i]&&map[pos][i]!=0x3f3f3f3f)
			{
				dis[i]=map[pos][i];
			}
		}
	}
	int temp=0;
	for(i=1;i<=N;i++)
	{
		temp+=dis[i];
	}
	printf("%d\n",temp);
}

int main()
{
	while(scanf("%d",&N),N)
	{
		M=N*(N-1)/2;
		memset(map,0x3f,sizeof(map));
		memset(visit,0,sizeof(visit));
		int a,b,val,flag;
		for(int i=1;i<=M;i++)
		{
			scanf("%d%d%d%d",&a,&b,&val,&flag);
			if(flag==1)
			{
				map[a][b]=0;
				map[b][a]=0;
			}
			else
			{
				if(val<map[a][b])
				{
					map[a][b]=val;
					map[b][a]=val;
				}
			}
		}
		prim();
	}
	return 0;
}