有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;
}