POJ 1861 Network(Kruskal)

传送门：http://poj.org/problem?id=1861

这题就是普通的最小生成树水题，第一次写最小生成树，1A了还是很开心的！~

这题是special judge，所以样例没过就交了，用一个栈来保存使用过的边，然后返回的是使用过的最大边。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
#include<iostream>
#include<cstring>
#include<stack>
using namespace std;
struct node{
	int s,e;
	int v;
	node(int a,int b,int c):s(a),e(b),v(c){}
};
bool cmp(node a,node b)
{
	return a.v<b.v;
}
vector<node> edge;
stack<int> st;
int pre[1010];
int Find(int x)//找根节点+压缩路径
{
	int s;
	for(s=x;pre[s]>=0;s=pre[s]);
	while(s!=x)
	{
		int t=pre[x];
		pre[x]=s;
		x=t;
	}
	return s;
}
void join(int a,int b)//并查集，将节点数少的集合并到节点数多的集合中去，能节约时间
{
	int fa=Find(a),fb=Find(b);
	int sum=pre[fa]+pre[fb];
	if(pre[fa]>pre[fb])
	{
		pre[fa]=fb;
		pre[fb]=sum;
	}
	else{
		pre[fb]=fa;
		pre[fa]=sum;
	}
}
int kruskal(int n)
{
	int ans=0;
	int i,j;
	int cnt=0;//当前使用的边数
	for(i=0;i<edge.size();i++)//将边按权值从小到大排序了，所以贪心取边，然后判断是否在同一个集合就可以了
	{
		if(Find(edge[i].s)!=Find(edge[i].e))
		{
			join(edge[i].s,edge[i].e);
			cnt++;
			ans=max(edge[i].v,ans);
			st.push(i);
		}
		if(cnt==n-1)
			break;
	}
	return ans;
}
int main()
{
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		while(!st.empty())
			st.pop();
		edge.clear();
		memset(pre,-1,sizeof(pre));
		int i,j;
		for(i=0;i<m;i++)
		{
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			edge.push_back(node(a,b,c));
		}
		sort(edge.begin(),edge.end(),cmp);
		printf("%d\n",kruskal(n));
		printf("%d\n",st.size());
		while(!st.empty())
		{
			printf("%d %d\n",edge[st.top()].s,edge[st.top()].e);
			st.pop();
		}
	}
	return 0;
}