HDU 1879 (最小生成树)

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
int cnt=0;
struct node{
	int a,b,tag;
	int cost;
}Node[5010];
int cmp (const node &a,const node &b){
	return a.cost < b.cost;
}

//----并查集操作----
int bleg[102];
int find (int x){
	int y=x;
	while (y != bleg[y])
		y = bleg[y];
	while (x != bleg[x]){
		int px = bleg[x];
		bleg[x] = y;
		x = px;
	}
	return y;
}
void Union (int a,int b){
	int pa=find (a),pb=find(b);
	if (pa != pb)   //非回路统计
		cnt++;
	if (rand() & 1)
		bleg[pa] = pb;
	else
		bleg[pb] = pa;
}

void init (int n){
	for (int i=1;i<=n;i++)
		bleg[i] = i;
}
//===========================
/*
 1.把已经存在道路的两个村庄加入到一个集合v中
 2.对所有的道路的长短从小到大排序
 3.筛选出没有道路联通并且也没有间接联通的节点i加入到集合v中。（可以确定i是最小的）
 4.重复3步骤直到v中有n-1条边或没有节点可加。
*/
int main (){
	#ifndef ONLINE_JUDGE
		freopen ("in.txt","r",stdin);
	#endif
	int n;
	while (~scanf ("%d",&n) && n){
		int a,b,cost,tag;
		int len = n*(n-1)/2;
		cnt = 0;
		init (n);
		for (int i=0;i<len;i++){
			scanf ("%d%d%d%d",&a,&b,&cost,&tag);
			Node[i].a=a,Node[i].b=b,Node[i].cost=cost,Node[i].tag=tag;
			if (Node[i].tag)  //如果i节点的a b两个村庄之间存在道路
				Union (Node[i].a,Node[i].b);  //那么他们就已经 连在一起。
		}
		sort (Node,Node+len,cmp);
		int sum = 0;bool flag=false;
		//--------------
		for (int i=0;i<len;i++){
			if (cnt==n-1) break;
		//	if (!Node[i].tag) //若i节点中的a b两个村庄不存在直接的道路
				if (find(Node[i].a) == find(Node[i].b)) //但他们之间是间接相通的
					continue;
				else {
					Union (Node[i].a,Node[i].b);
					sum += Node[i].cost;
				}
		}
	//	printf ("flag == %d\n",flag);
		if (cnt==n-1)
			printf ("%d\n",sum);
		else
			puts ("0");
	}
	return 0;
}

wa了多次的代码：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
int cnt=0;
struct node{
	int a,b,tag;
	int cost;
}Node[5010];
int cmp (const node &a,const node &b){
	return a.cost < b.cost;
}

//----并查集操作----
int bleg[102];
int find (int x){
	int y=x;
	while (y != bleg[y])
		y = bleg[y];
	while (x != bleg[x]){
		int px = bleg[x];
		bleg[x] = y;
		x = px;
	}
	return y;
}
void Union (int a,int b){
	int pa=find (a),pb=find(b);
	if (pa != pb)   //非回路统计
		cnt++;
	if (rand() & 1)
		bleg[pa] = pb;
	else
		bleg[pb] = pa;
}

void init (int n){
	for (int i=0;i<n;i++)
		bleg[i] = i;
}
//===========================
/*
 1.把已经存在道路的两个村庄加入到一个集合v中
 2.对所有的道路的长短从小到大排序
 3.筛选出没有道路联通并且也没有间接联通的节点i加入到集合v中。（可以确定i是最小的）
 4.重复3步骤直到v中有n-1条边或没有节点可加。
*/
int main (){
	#ifndef ONLINE_JUDGE
		freopen ("in.txt","r",stdin);
	#endif
	int n;
	while (~scanf ("%d",&n) && n){
		int a,b,cost,tag;
		int len = n*(n-1)/2;
		cnt = 0;
		init (n);
		for (int i=0;i<len;i++){
			scanf ("%d%d%d%d",&a,&b,&cost,&tag);
			Node[i].a=a,Node[i].b=b;  //=======bleg是错位的的，但是a b并没有错位 -1。
			Node[i].cost=cost,Node[i].tag=tag;
			if (Node[i].tag)  //如果i节点的a b两个村庄之间存在道路
				Union (Node[i].a,Node[i].b);  //那么他们就已经 连在一起。
		}
		sort (Node,Node+len,cmp);
		int sum = 0;bool flag=false;
		//--------------
		for (int i=0;i<len;i++){
			if (cnt==n-1) break;
		//	if (!Node[i].tag) //若i节点中的a b两个村庄不存在直接的道路
				if (find(Node[i].a) == find(Node[i].b)) //但他们之间是间接相通的
					continue;
				else {
					Union (Node[i].a,Node[i].b);
					sum += Node[i].cost;
				}
		}
	//	printf ("flag == %d\n",flag);
		if (cnt==n-1)
			printf ("%d\n",sum);
		else
			puts ("0");
	}
	return 0;
}