CODJ 1070 秋实大哥打游戏 并查集

有N个独立的点，标号为1，2，3，…N。现在要把他们连接起来，每次选取x和y两个点，x是它所处集合的中心，y不一定是它所处集合的中心，然后在xy间连一条长度为|x-y|mod1000的边。然后把这两个集合合并起来，新集合的中心是原y所处的集合的中心。有两种操作，I x y表示选取x和y合并。E x表示查询x到它所处集合中心的距离。O表示结束。

并查集的套路，但是在查询的时候，路径压缩的时候，更新点到集合中心的距离即可。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <string>
using namespace std;
#define maxn 100005
int set[maxn];
//int size[maxn];//结点数
int dis[maxn];
//int center[maxn];
int N;
void init_set()
{
	memset(dis, 0, sizeof(int)*maxn);
	for (int i = 0; i<maxn; ++i)
	{
		set[i] = i;
		//size[i] = 1;
		//center[i] = 1;
	}
}
int findSet(int x)//路径压缩
{
	if (x == set[x])
		return x;
	else
	{
		/*int fx = set[x];
		dis[x] += dis[fx];
		set[x] = set[fx];
		dis[x] += dis[fx];*/
		int fx = set[x];
		int rx = findSet(fx);
		dis[x] += dis[fx];
		set[x] = rx;
		return rx;
	}
}
void unionSet(int x, int y)
{
	//int fx = findSet(x);
	int fy = findSet(y);
	if (fy == x)
		return;
	set[x] = fy;
	dis[x] = dis[x] + abs(x - y) % 1000 + dis[y];
}
int main()
{
	//freopen("input.txt", "r", stdin);
	scanf("%d", &N);
	init_set();
	char cmd[30];
	memset(cmd, 0, sizeof(char) * 30);
	while (gets(cmd))
	{
		if (cmd[0] == 'O')
			break;
		else if (cmd[0] == 'I')
		{
			int x = 0, y = 0;
			int i;
			for (i = 1; !(cmd[i] >= '0'&&cmd[i] <= '9'); ++i);
			for (; cmd[i] >= '0'&&cmd[i] <= '9'; ++i)
				x = x * 10 + cmd[i] - '0';
			for (; !(cmd[i] >= '0'&&cmd[i] <= '9'); ++i);
			for (; cmd[i] >= '0'&&cmd[i] <= '9'; ++i)
				y = y * 10 + cmd[i] - '0';
			unionSet(x, y);
		}
		else if (cmd[0] == 'E')
		{
			int x = 0, i;
			for (i = 1; !(cmd[i] >= '0'&&cmd[i] <= '9'); ++i);
			for (; cmd[i] >= '0'&&cmd[i] <= '9'; ++i)
				x = x * 10 + cmd[i] - '0';
			findSet(x);
			printf("%d\n", dis[x]);
		}
		memset(cmd, 0, sizeof(char) * 30);
	}
	//system("pause");
	//while (1);
	return 0;
}