【BZOJ 1202】【HNOI 2005】 狡猾的商人

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

---

题解：

这题莫名的并查集，真是太妙了！！！
类似于前缀和，

对于一组数据{l,r,w}，相当于求s[r]-s[l-1]；同时，如果知道了s[a]-s[b]和s[b]-s[c]就可以知道s[a]-s[c]。

所以，对于{l,r,w},我们将l和r放入一个并查集，并维护每个点到root的数值（就是前缀和中的第一个元素）即可。

若l和r已经在同一个并查集中，就不断查询求值。

在寻找root的时候可以顺便维护值。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;

#define N 1010
int f[N],v[N];
int n,m,t;
bool w;


int find(int x)
{
    if(f[x] == x) return x;
    int q = find(f[x]);
    v[x] += v[f[x]];
    f[x] = q;
    return f[x];
}

void add(int x,int y,int z)
{
    int p = find(x),q = find(y);
    if(p != q)
    {
        f[p] = q;
        v[p] = v[y] - v[x] - z;
    }
    else if(v[y] - v[x] != z) w = 1;
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        memset(v,0,sizeof(v));
        scanf("%d%d",&n,&m);
        w = 0;
        for(int i = 0;i <= n;i++) f[i] = i;
        for(int i = 1;i <= m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            if(!w) add(x-1,y,z);
        }
        if(w) printf("false\n");
        else printf("true\n");
    }
    return 0;
}