[HNOI2005][BZOJ1202] 狡猾的商人

1202: [HNOI2005]狡猾的商人

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1863  Solved: 899
[Submit][Status][Discuss]

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

 

想法好像比较奇怪……打死我也没想到这是个并查集啊。

并查集维护两个数组，f[i]表示i所在的并查集，v[i]表示f[i]->i月的收入，每次读入s,t,令p=f[s],q=f[t],若p==q，说明s,t属于同一个并查集，找f[s],f[t]的过程中维护处v[s],v[t]，flag=v[t+1]-v[s]==w?1:0。若p!=q，则将f[q]赋值为p，（令每个并查集的左端点作为父亲结点，维护v[q]）。

另：由于维护的是前缀和，所以读入是可将t+1，判断v[t+1]-v[s]的值。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int f[110],v[110];
int n,m,s,t,w,ww;
bool flag;
int find(int x)
{
    if (f[x]==x) return x;
    int pp=f[x];
    f[x]=find(f[x]);
    v[x]+=v[pp];
    return f[x];
}
int main()
{
    scanf("%d",&w);
    while (w--)
    {
        memset(v,0,sizeof(v));
        flag=0;
        scanf("%d%d",&n,&m);
        for (int i=1;i<=n+1;i++) f[i]=i;
        for (int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&s,&t,&ww);
            t++;
            int p=find(s),q=find(t);
            if (p==q)
            {
                if (v[t]-v[s]!=ww) 
                {
                    flag=1;
                    break;
                }
            }
            else
            {
                f[q]=p;
                v[q]=v[s]+ww-v[t];
            }
        }
        if (flag) printf("false\n"); else printf("true\n");
    }
    return 0;
}