BZOJ 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

http://www.lydsy.com/JudgeOnline/problem.php?id=1202

原本题目里给的每个月是一个时间段，而将每次输入的右边界t++，就可以使它变成时间点

接着如果给了a-b的信息，又给了b-c或c-b的信息，就可以知道a-c或c-a的信息。

所以我们可以用加权并查集处理，对于每个点a用dis[a]记录s[fa]-s[a-1]root为并查集中a的父亲，所以就可以通过dis[a]-dis[b]，知道s[a]到s[b]的和，解决账本是否合理的问题。(路径压缩的特殊处理请看代码)

而当我们合并两个连通块时，将a所在连通块的根节点c变成b所在连通块的根节点d的儿子，同时记录dis[c]=dis[b]+v-dis[a];

dis[c]表示c到d，dis[b]表示b到d，v表示a到b,dis[a]表示a到c

#include<cstdio>
#include<cstring>
int fa[100005],dis[100005];
int fi(int a)
{
    if(!fa[a]) return a;
    int b=fi(fa[a]);
    dis[a]+=dis[fa[a]];
    return fa[a]=b;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m,sf=1;
        scanf("%d%d",&n,&m);
        memset(fa,0,sizeof(fa));
        memset(dis,0,sizeof(dis));
        for(int i=1;i<=m;++i)
        {
            int l,r,a;
            scanf("%d%d%d",&l,&r,&a);
            ++r;
            int t1=fi(l),t2=fi(r);
            if(t1-t2)
            {
                fa[t1]=t2;
                dis[t1]=a+dis[r]-dis[l];
            }
            else if(dis[l]-dis[r]!=a) sf=0;
        }
        sf?printf("true
"):printf("false
");
    }
    return 0;
}