BZOJ 1202 [带权并查集]

1202: [HNOI2005]狡猾的商人
Description
刁姹接到一个任务，为税务部门调查一位商人的账本，看看账本是不是伪造的。账本上记录了n个月以来的收入情况，其中第i 个月的收入额为Ai(i=1,2,3…n-1,n)， 。当 Ai大于0时表示这个月盈利Ai 元，当 Ai小于0时表示这个月亏损Ai 元。所谓一段时间内的总收入，就是这段时间内每个月的收入额的总和。 刁姹的任务是秘密进行的，为了调查商人的账本，她只好跑到商人那里打工。她趁商人不在时去偷看账本，可是她无法将账本偷出来，每次偷看账本时她都只能看某段时间内账本上记录的收入情况，并且她只能记住这段时间内的总收入。 现在，刁姹总共偷看了m次账本，当然也就记住了m段时间内的总收入，你的任务是根据记住的这些信息来判断账本是不是假的。
Input
第一行为一个正整数w，其中w < 100，表示有w组数据，即w个账本，需要你判断。每组数据的第一行为两个正整数n和m，其中n < 100，m < 1000，分别表示对应的账本记录了多少个月的收入情况以及偷看了多少次账本。接下来的m行表示刁姹偷看m次账本后记住的m条信息，每条信息占一行，有三个整数s，t和v，表示从第s个月到第t个月（包含第t个月）的总收入为v，这里假设s总是小于等于t。
Output
包含w行,每行是true或false,其中第i行为true当且仅当第i组数据，即第i个账本不是假的；第i行为false当且仅当第i组数据，即第i个账本是假的。
Sample Input
2
3 3
1 2 10
1 3 -5
3 3 -15
5 3
1 5 100
3 5 50
1 2 51
Sample Output
true
false

刚开始还想是判断某点到其他点的最长路和最短路是否相同。。。。
反正最后想到了带权并查集。
每次得到三个数x, y, z 因为是[x, y]，为了具有普遍性 将其改写成[x, y + 1)，那么就可以这么理解y + 1就是x的父节点。权值为z
判断错误很简单 如果 find(x) == find(y) && key[x] != key[y] + z（这里是加号，要注意，因为key[x]表示x到祖先的距离，如图）

先上找爸爸函数：

inline int find(int x) {
  if (f[x] == x) return x;
  int fa = f[x];
  f[x] = find(fa);
  key[x] += key[fa];//带权并查集的合并
  return f[x];
}

合并要分两种情况：
一种是 $F_x$ < $F_y$，那么还是看图吧

很容易求到图中问号的值为key[y] + z - key[x]
另一种 $F_x$ > $F_y$，如图

同理可以求出问号值为key[x] - z - key[y]
那么最后这道题就解决了

附代码：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#define N 110
using namespace std;

int test, n, m, x, y, z;
int f[N], key[N];

inline int find(int x) {
  if (f[x] == x) return x;
  int fa = f[x];
  f[x] = find(fa);
  key[x] += key[fa];
  return f[x];
}

int main(void) {
  scanf("%d", &test);
  while (test--) {
    scanf("%d%d", &n, &m);
    bool flag = 1;
    for (int i = 1; i <= n + 1; i++) f[i] = i, key[i] = 0;
    for (int i = 0; i < m; i++) {
      scanf("%d%d%d", &x, &y, &z); y++;
      int fx = find(x), fy = find(y);
      if (fx != fy) {
        if (fx < fy) {
          f[fx] = fy; key[fx] = key[y] + z - key[x];
        } else {
          f[fy] = fx; key[fy] = key[x] - z - key[y];
        }
      } else {
        if (key[x] - z != key[y]) {
          for (i++; i < m; i++) scanf("%d%d%d", &x, &x, &x);
          flag = 0; puts("false"); break;
        }
      }
    }
    if (flag) puts("true");
  }
  return 0;
}