畅通工程(MST,存在得不到MST的情况)

题目描述

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

输入描述:

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

输出描述:

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

输入

3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

输出

3
?
注意:该例存在得不到最小生成树的情况,所以最后我们 对所有节点是否属于同一个集合进行判断。
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define N 1000
int Tree[N];
struct edge{
	int a,b;
    int cost ;
    bool operator< (const edge &A)const{
        return cost<A.cost;
    }
}edge[6000];
int findroot(int x)//查找根节点
{
    if(Tree[x]==-1)
        return x;
    else{
        int tmp=findroot(Tree[x]);
        Tree[x]=tmp;
        return tmp;
    }
}
int main()
{
    int n,m,i,j;
    while(~scanf("%d%d",&n,&m)&&n!=0)
    { 
        for( i=1;i<=n;i++)
        {
            scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].cost);
        }
        sort(edge+1,edge+1+n);
        for(i=1;i<=m;i++)Tree[i]=-1;
		
        int ans=0; int a,b;
        for( i=1;i<=n;i++){
			
            a=findroot(edge[i].a);
            b=findroot(edge[i].b);
            if(a!=b){
                Tree[a]=b;
                ans+=edge[i].cost;
            }
        } 
		if(m==1)
			printf("%d\n",ans ); 
		else
		{ 
			a=findroot(1);
			for(i=2;i<=m;i++)
			{
				b=findroot(i); 
				if(a!=b)
				{printf("?\n");break;}
			}
			if(i==m+1)
				printf("%d\n",ans );  
		}
		
    }
    return 0;
}



猜你喜欢

转载自blog.csdn.net/sunshine_lyn/article/details/79603194
MST