POJ-2236 并查集

不知道STATISTICS中第一页的程序是怎么写出来的。

贴上我的950ms+的代码。

/*
 * wireless network
 * mike-w
 * 2012-4-28
 */
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>

#define MAX_SIZE 1234

double co[MAX_SIZE][2]; /* coordinate */
int flag[MAX_SIZE];
int set[MAX_SIZE],rank[MAX_SIZE];
int N;
double D;

int set_init(int size)
{
	int i;
	for(i=1;i<=size;i++)
		set[i]=i,rank[i]=1;
	return 0;
}

int set_find(int e)
{
	if(set[e]==e)
		return e;
	else
		return set[e]=set_find(set[e]);
}

int set_merge(int e1, int e2)
{
	int r1=set_find(e1);
	int r2=set_find(e2);
	if(r1==r2)
		return -1;
	if(rank[r1]>rank[r2])
		set[r2]=r1;
	else if(rank[r1]<rank[r2])
		set[r1]=r2;
	else 
		set[r1]=r2,rank[r2]++;
	return 0;
}

int link(int node)
{
	int i,dx,dy;
	flag[node]=1;
	for(i=1;i<=N;i++)
	{
		if(flag[i])
		{
			dx=co[i][0]-co[node][0];
			dy=co[i][1]-co[node][1];
			if(sqrt(dx*dx+dy*dy)<=D)
				set_merge(i,node);
		}
	}
	return 0;
}

int main(void)
{
#ifndef ONLINE_JUDGE
	freopen("in","r",stdin);
#endif
	int i,e1,e2;
	char buf[5];

	scanf("%d%lf",&N,&D);
	set_init(N);
	memset(flag,0,sizeof(flag));
	for(i=1;i<=N;i++)
		scanf("%lf%lf",co[i],co[i]+1);
	while(scanf("%s",buf)!=EOF)
	{
		if(buf[0]=='O')
			scanf("%d",&e1),link(e1);
		else
		{
			scanf("%d%d",&e1,&e2);
			puts(set_find(e1)==set_find(e2)?"SUCCESS":"FAIL");
		}
	}
	
	return 0;
}