HDU 4829 Information 带权并查集 模拟

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=4829

题意

一道有6个操作的模拟题：
军情紧急，我们需要立刻开发出一个程序去处理前线侦察兵发回的情报，并做出相应的分析。现在由你负责其中的一个子模块，你需要根据情报计算出敌方坦克的位置。
当敌方坦克静止时，侦察兵会尽力估算出它们之间的位置，而每当敌方坦克移动时，侦察兵都会记录下坦克新的位置并向你报告。每个坦克的位置可以由一个二维整数坐标来描述。
前线发回的情报有四种格式：
1 A B X Y
表示A坦克移动到了与B坦克的相对位置是(X,Y)的地方，即XA = XB + X, YA=YB+Y。
2 A X Y
表示A坦克移动到了绝对位置是(X,Y)的地方，即XA = X, YA = Y。
3 A B X Y
表示发现了A坦克与B坦克的相对位置是(X,Y)，即XA = XB + X, YA=YB+Y。
4 A X Y
表示发现了A坦克的绝对位置是(X,Y)，即XA = X, YA = Y。
我们需要你对于如下两种询问及时做出回应：
5 A B
表示询问A坦克与B坦克的相对位置是多少，即分别求出XA - XB 以及YA -YB。
6 A
表示询问A坦克的绝对位置是多少，即求出XA 和YA。
其中A和B代表的是任意的一个坦克的编号，(X,Y)表示了坦克的二维坐标。你可以假设初始时刻我们对于敌方任何坦克的位置都一无所知，在此之后坦克的每一次移动都被侦察兵侦察到了。
请注意两个坦克的坐标有可能相同。

对于每组数据，首先需要输出单独一行”Case #?:”，其中问号处应填入当前的数据组数，组数从1开始计算。
对于每一个类型(1)或者(2)的询问，请把它们加入到你的记录中。
对于每一个类型(3)或者(4)的询问，如果与之前记录的内容有矛盾，请输出”REJECT”并将这个情报忽略掉，如没有矛盾，请把它们加入到你的记录中。
对于每一个类型(5)或者(6)的询问，如果根据之前的记录能推出结论，请输出两个整数X和Y，两个整数之间有一个空格；如果不能推出结论，请输出”UNKNOWN”。输出的所有信息都不包括引号。

思路：

用带权并查集进行纯模拟，好久没做这种题了，后来看了别人的题解才会，考虑到这篇题解我百度不到 = =（百度到的题解有问题），自己也写一写把。

首先模拟题最好是分析清楚题意，捋清各个操作之间的异同，这样代码写起来简短易读，不然一个又一个的if else让人看了头大= =

带权并查集是自己第一次接触，每个点多了一个d[]数组，用来记录这个点到根节点的距离。

废话了一通，这题的关键点在于，1，2操作会把一个点从原有点集中拿出，这样很多点的相对关系就会乱，所以我们每次不把原有点从点集拿出，而是抛弃原有的点，用新的点代替，这样原有点集之间的相对关系就不会乱掉

代码：

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#include <queue>using namespace std;typedef pair<int, int> P;const int MAXN = 2e5 + 5;P operator+(const P &a, const P &b) { return P(a.first + b.first, a.second + b.second); }P operator-(const P &a, const P &b) { return P(a.first - b.first, a.second - b.second); }bool operator!=(const P &a, const P &b) { return a.first != b.first || a.second != b.second; }int n;int pa[MAXN], r[MAXN];int node_num;P d[MAXN];int findset(int x) {  if (pa[x] != x) {    int root = findset(pa[x]);    d[x] = d[x] + d[pa[x]];    return pa[x] = root;  } else return x;}int newnode() {  pa[node_num] = node_num;  d[node_num] = P(0, 0);  return node_num++;}int main() {//  freopen("in.txt", "r", stdin);//  freopen("out.txt", "w", stdout);  int T, kase = 0;  scanf("%d", &T);  while (T--) {    printf("Case #%d:\n", ++kase);    scanf("%d", &n);    node_num = 0;    for (int i = 0; i <= n; ++i) r[i] = newnode();//    for (int i = n + 1; i <= n << 1; ++i) d[i] = P(0, 0);    int op, a, b, x, y, now, fa, fb;    for (int i = 1; i <= n; ++i) {      scanf("%d%d", &op, &a);      b = 0;      if (op == 1 || op == 2) {        if (op == 1) scanf("%d%d%d", &b, &x, &y);        else scanf("%d%d", &x, &y);        r[a] = newnode();        a = r[a]; b = r[b];        fb = findset(b);        pa[a] = fb;        d[a] = P(x, y) + d[b];      } else if (op == 3 || op == 4) {        if (op == 3) scanf("%d%d%d", &b, &x, &y);        else scanf("%d%d", &x, &y);        a = r[a]; b = r[b];        fa = findset(a); fb = findset(b);        if (fa == fb) {          if ((d[a] - d[b]) != P(x, y)) puts("REJECT");        } else {          pa[fa] = fb;          d[fa] = d[b] - d[a] + P(x, y);        }      } else if (op == 5 || op == 6) {        if (op == 5) scanf("%d", &b);        a = r[a]; b = r[b];        fa = findset(a); fb = findset(b);        if (fa != fb) puts("UNKNOWN");        else printf("%d %d\n", d[a].first - d[b].first, d[a].second - d[b].second);      }    }  }}