还是畅通工程 hdu1233（Kruskal算法）模板题 TLE改法

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 73901    Accepted Submission(s): 33376

 

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

 

Sample Input

3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0

 

 

Sample Output

3 5

Hint

Hint

Huge input, scanf is recommended.

 

 

Source

浙大计算机研究生复试上机考试-2006年

---

我的并查集有点长，所以说代码会有点长.下一题我用Prim去做顺便优先队列优化一下

等下把Dijkstra也用优先队列优化一下

TLE的把数组开到10000+

---

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=10005;
int par[maxn];
int height[maxn];

struct edge
{
	int u;
	int v;
	int cost;
}es[maxn];

bool comp(const edge& e1,const edge&e2)
{
		return e1.cost<e2.cost;
}

void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		par[i]=i;
		height[i]=0;
	}
}
int find(int x)
{
	if(par[x]==x)
	{
		return x;
	}
	else{
		return par[x]=find(par[x]);
		
	}
}
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y) return;
	
	if(height[x]<height[y])
	{
		par[x]=y;
		
	}
	else
	{
		par[y]=x;
		if(height[x]==height[y])
		height[x]++;
	}
}
bool same(int x,int y)
{
	return find(x)==find(y);
}


int V,E;
int kruslkal()
{
	sort(es+1,es+E+1, comp);
	//init(V);/// 
	
	int res=0;
	for(int i=1;i<=E;i++)
	{
		edge e=es[i];
		if(!same(e.u,e.v))
		{
			unite(e.u,e.v);
			res+= e.cost;
		}
	}
return res;	
}
int main(void)
{
	while(scanf("%d",&V)!=EOF)
	{
		if(V==0) break;
		
		memset(par,0,sizeof(par));memset(height,0,sizeof(height));memset(es,0,sizeof(es));
		init(V);
		E=V*(V-1)/2; 
		for(int i=1;i<=E;i++)
		{
		 scanf("%d%d%d", &es[i].u, &es[i].v, &es[i].cost); 
		}	
		int ans=kruslkal();
		printf("%d\n",ans);
	}
return 0;
}