BZOJ 1202: [HNOI2005]狡猾的商人 带权并查集

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 3960 Solved: 1909

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

HINT

Source

---

23333
dalao讲解

---

这题好神啊，又学到个神奇操作，带权并查集
首先我们研究并查集的本质，发现其实并查集是一棵树，最开始我们的并查集都是一个单点，然后慢慢地联通的，于是我们维护一个神奇的值是从这个点到根节点的差值，我们一会儿再来讨论维护的这个值，现在我们先看这个：
输入了 $l,r,num$，显然我们现在是$l$到$r$的值之和为$num$，如果我们把一个点一个点的前缀和表示为$s$，则$s[r] - s[ l - 1 ] = num$，因此我们可以维护一个值表示这个值到他的根的差值，设这个差值数组是$sum$，而如果$l-1$这个点和$r$这个点是联通的，那么显然有$num = sum[l - 1] - sum[r]$因为我们的$sum$数组表示的是差值，也就是说从根节点到”最左边的节点”的距离减去从$l-1$到”最左边的节点”的距离的值，那么$l - 1$的这段值如果减去$r$的这段值如果不等于输入的$num$，也就是之前输出的内容有冲突，则说明$false$，如果没有冲突则在这条账目上是对的
如果两个点的$fa$不同，则需要新维护

$$s[y] - s[x] = num$$
$$s[fax]=sum[x]+s[x]$$
$$s[fay]=sum[y]+s[y]$$

因此我们可以直接根据这个得出新的fay这个节点到根节点的差值，为$s[fax]-s[fay]$，发现$s[y]-s[x]=num$
于是

$$sum[fay]=sum[x]-sum[y]-num$$

---

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MAXN = 1000;
int fa[MAXN], T, n, m, sum[MAXN];
int find( int x ) {
    if( x == fa[x] ) return x;
    int fax = find( fa[x] );
    sum[x] += sum[fa[x]];
    fa[x] = fax;
    return fa[x];
}
int main( ) {
    scanf( "%d", &T ); 
    while( T-- ) { bool judge = true;
        scanf( "%d%d", &n, &m );
        for( register int i = 0; i <= n; i++ ) fa[i] = i;
        memset( sum, 0, sizeof(sum) );
        for( register int i = 1; i <= m; i++ ) { int l, r, num;
            scanf( "%d%d%d", &l, &r, &num ); l--;
            int fal = find(l); int far = find(r);
            if( fal == far ) {
                if( ( sum[l] - sum[r]  ) == num ) continue;
                else { judge = false; break; }
            } else {
                sum[far] = sum[l] - sum[r] - num;
                fa[far] = fal;
            }
        }
        if( judge ) printf( "true\n" );
        else        printf( "false\n" );
    }
}