BZOJ-1202 狡猾的商人 并查集+前缀和

我记得这个题，上次之前做的时候没改完，撂下了，今天突然想改发现，woc肿么A 了= =看来是我记错了。。

1202: [HNOI2005]狡猾的商人
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 2232 Solved: 1060
[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

并查集+前缀和维护。似乎没什么好说的。。。（其实做的太早似乎快忘 了。。）

code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int father[10000]={0};
int dis[10000]={0};
bool flag=true;
int w,n,m;
int x,y,z;

int find(int x)
{
  if (father[x]==x) 
   return x;
  int temp=find(father[x]);
  dis[x]+=dis[father[x]];
  father[x]=temp;
  return father[x];
}

void merge(int x,int y)
{
    int f1=find(x);
    int f2=find(y);
    if (f1!=f2)
     {
        father[f1]=f2;
        dis[f1]=dis[y]-dis[x]-z;        
     }
    else
     if (dis[y]-dis[x]!=z)
      flag=false;
}

int main()
{
    scanf("%d",&w);
    for (int t=1;t<=w;t++)
     {
        scanf("%d%d",&n,&m);
        for (int i=0;i<=n;i++)
         father[i]=i;
        memset(dis,0,sizeof(dis));
        flag=true;
        for (int i=1;i<=m;i++)
         {
            scanf("%d%d%d",&x,&y,&z);
            merge(x-1,y);
         }
        if (flag)
         printf("%s
","true");
        else
         printf("%s
","false");
     }
  return 0;
}