A - 畅通工程(并查集)

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。

Output

对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。

Sample Input

3 3

1 2 1

1 3 2

2 3 4

1 3

2 3 2

0 100

Sample Output

3

?

  • 题意概括  :

M个城镇,n条道路,每条道路给出两个村庄的编号以及村庄之间道路的成本,求使所有城镇保持畅通的最小成本,如果不能使所有城镇保持畅通输出?。

  • 解题思路  :

Krukal算法末班,同时加入一个数字count判断所加入树中的点的个数,同时用sum记录所每条道路的权值和,如果树中的点达到n-1个,则证明所有城镇保持畅通输出sum,如果没有则输出?。

#include<stdio.h>
#include<string.h>
#include<algorithm>

using namespace std;

struct node
{
	int u,v,w;
}e[110],t;

int n,m,f[110],sum;

void quicksort(int left,int right)
{
	int i,j;
	
	if(left > right)
	return;
	
	i = left;
	j = right;
	
	while(i!=j)
	{
		while(e[j].w >= e[left].w && i<j)
		{
			j --;
		}
		while(e[i].w <= e[left].w &&i<j)
		{
			i ++;
		}
		if(i < j)
		{
			t = e[i];
			e[i] = e[j];
			e[j] = t;
		}
	}
	t = e[left];
	e[left] = e[i];
	e[i] = t;
	
	quicksort(left,i-1);
	quicksort(i+1,right);
	return;
}

int getf(int v)
{
	if(f[v] != v)
	f[v] = getf(f[v]);
	return f[v];
}
int merge(int v,int u)
{
	int t1,t2;
	
	t1 = getf(v);
	t2 = getf(u);
	
	if(t1!=t2)
	{
		f[t2] = t1;
		return 1;
	}
	return 0;
}

int main()
{
	int i,flag,count;
	
	while(~ scanf("%d %d",&m,&n))
	{
		if(m == 0)
		break;
		count = 0;
		sum = 0;
		for(i = 1;i<=m;i ++)
		{
			scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].w);
		}
		quicksort(1,m);
		for(i = 1;i<=n;i ++)
		{
			f[i] = i;
		}
		
		for(i = 1;i<=m;i ++)
		{
			if(merge(e[i].u,e[i].v))
			{
				count ++;
				sum += e[i].w;
			}
			
			if(count == n-1)
			{
				break;
			}
		}
		if(count == n-1)
		printf("%d\n",sum);
		else
		printf("?\n");
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/y1356998843/article/details/81511584