P2294 [HNOI2005]狡猾的商人

题目描述

***姹接到一个任务，为税务部门调查一位商人的账本，看看账本是不是伪造的。账本上记录了n个月以来的收入情况，其中第i 个月的收入额为Ai(i=1,2,3…n-1,n)， 。当 Ai大于0时表示这个月盈利Ai 元，当 Ai小于0时表示这个月亏损Ai 元。所谓一段时间内的总收入，就是这段时间内每个月的收入额的总和。 ***姹的任务是秘密进行的，为了调查商人的账本，她只好跑到商人那里打工。她趁商人不在时去偷看账本，可是她无法将账本偷出来，每次偷看账本时她都只能看某段时间内账本上记录的收入情况，并且她只能记住这段时间内的总收入。 现在，***姹总共偷看了m次账本，当然也就记住了m段时间内的总收入，你的任务是根据记住的这些信息来判断账本是不是假的。

输入格式

第一行为一个正整数w，其中w < 100，表示有w组数据，即w个账本，需要你判断。每组数据的第一行为两个正整数n和m，其中n < 100，m < 1000，分别表示对应的账本记录了多少个月的收入情况以及偷看了多少次账本。接下来的m行表示***姹偷看m次账本后记住的m条信息，每条信息占一行，有三个整数s，t和v，表示从第s个月到第t个月（包含第t个月）的总收入为v，这里假设s总是小于等于t。

输出格式

包含w行,每行是true或false,其中第i行为true当且仅当第i组数据，即第i个账本不是假的；第i行为false当且仅当第i组数据，即第i个账本是假的。

输入输出样例

输入 #1
2
3 3
1 2 10
1 3 -5
3 3 -15
5 3
1 5 100
3 5 50
1 2 51
输出 #1
true
false

思路

先做前缀和，用并查集来判断是真是假。。。

一定一定要记得每次循环前把flag赋为0！！！！！！！！！

代码

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

const int N=10010;

int tmp,flag;
int a[N],f[N];
int w,n,m,s,t,v;

int find(int x) {
	if(x==f[x])
		return x;
	tmp=find(f[x]);
	a[x]+=a[f[x]];
	f[x]=tmp;
	return f[x];
}

void check(int x,int y,int w) {
	int p=find(x),q=find(y);
	if(p==q) {
		if(a[y]-a[x]!=w)
			flag=1;
	} else {
		f[p]=q;
		a[p]=a[y]-a[x]-w;
	}
}

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