hiho coder 最小生成树

简直蛋疼

简直————

不说了，直接上代码。

krustral算法，并查集避免环。

AC代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int n, m;
const int MAX = 1000005;
struct edge
{
	int a, b, w;
}e[MAX];
int p[MAX];

int cmp(edge a, edge b)
{
	return a.w < b.w;
}

int find(int x)
{
	if(p[x] != x) p[x] = find(p[x]);
	return p[x];
}

int join(edge e)
{
	int x = find(e.a), y = find(e.b);
	if(x != y)
	{
		p[x] = y;
		return e.w;
	}
	return 0;
}

int kruskal()
{
	int ans = 0;
	for(int i=1; i<=n; i++) p[i] = i;
	sort(e, e+m, cmp);
	for(int i=0; i<m; i++) ans += join(e[i]);
	return ans;
}

int main()
{
	while(scanf("%d %d", &n, &m) != EOF)
	{
		for(int i=0; i<m; i++)
			scanf("%d %d %d", &e[i].a, &e[i].b, &e[i].w);
		printf("%d\n", kruskal());
	}
	return 0;
}

RUNTIME ERROR

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
int f[100010];
struct Node
{
int a,b,c;
}node[1000000];
bool cmp(Node s1,Node s2)
{
return s1.c<s2.c;
}
int find(int x)
{
return x==f[x]?x:find(f[x]);
}
void init(int n)
{
for(int i=1;i<=n;i++)
f[i]=i;
}
void merge(int a,int b)
{
a=find(a);
b=find(b);
if(a!=b) f[a]=b;
}
int main()
{
int N,M;
scanf("%d%d",&N,&M);
init(N);
for(int i=0;i<M;i++)
scanf("%d%d%d",&node[i].a,&node[i].b,&node[i].c);
sort(node,node+M,cmp);
int ans=0;
int qqq=0;
for(int i=0;i<M;i++)
{
int aa=find(node[i].a);
int bb=find(node[i].b);
if(aa!=bb) { ans+=node[i].c;qqq++;f[aa]=bb;}
if(qqq==N) break;
}
printf("%d\n",ans);
return 0;
}