【Jungle Roads】【POJ - 1251】

题目链接 :http://poj.org/problem?id=1251

绝对是一道超级经典的最小生成树题目,有好多种解法。

解题思路:每次先找最小的边,取出看看将它放进去,是否有环生成,若没有,加入,反之,继续下一条比它稍微长点的边

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<stack>
#include<queue>
#define maxn 100
using namespace std;
int n,cnt,kase;
int far[maxn];
struct node{
	int u,v;
	int w;
}lu[maxn];
bool cmp(node a,node b)
{
	return a.w<b.w;
}
int find(int x)
{
	int r=x;
	while(r!=far[r])
		r=far[r];
	int i=r,j;
	if(i!=r)
	{
		j=far[i];
		far[i]=r;
		i=j;	
	}	
	return r;
} 
void merge(int a,int b)
{
	int pa=find(a);
	int pb=find(b);
	if(pa!=pb)
	{
		far[pb]=pa;
		cnt++;	
		kase=1;
	} 
}
int main()
{
	while(scanf("%d",&n)&&n)
	{
		int ans=0;
		int sum=0;
		cnt=0;
		for(int i=1;i<=n;i++)
			far[i]=i;
		for(int i=1;i<n;i++)
		{
			char a;
			int b;
			cin>>a;
			cin>>b;
			while(b--)
			{
				char c;
				lu[ans].u=i;
				cin>>c;
				lu[ans].v=c-'A'+1;
				cin>>lu[ans++].w;
			}
		}
		sort(lu,lu+ans,cmp);

		for(int i=0;i<ans;i++)
		{
			kase=0;
			merge(lu[i].u,lu[i].v);
			if(kase)
				sum+=lu[i].w;
		}
		if(cnt==n-1)
			cout<<sum<<endl;
	
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_42505741/article/details/81431446