BZOJ 1202: [HNOI2005]狡猾的商人 并查集维护前缀和

1202: [HNOI2005]狡猾的商人

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 3740  Solved: 1798
[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

Oakley~

用并查集维护类似前缀和的东西，用v[x]表示从x到连通块根的价值

如果两个点已经在一个集合里说明它们之间的价值已经求出来了，比如点对（x,y）已经在一个集合里,只需要判断v[y]-v[x]?=z；若不在一个集合里，将x，y合并，合并过程如下：

1.设p=find(x),q=find(y); 

2.v[y]=s[y]-s[q],v[x]=s[x]-s[p];z=s[y]-s[x];

3.得出v[p]=v[y]-v[x]-z;

不断用并查集维护v[]即可

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=110;

bool flag;

int v[N],fa[N];

inline int find(int x)
{
	if(x==fa[x])return x;
	int t=find(fa[x]);
	v[x]+=v[fa[x]];fa[x]=t;
	return t;
}

void judge(int x,int y,int w)
{
	if(flag)return ;
	int fax=find(x),fay=find(y);
	if(fax==fay){if(v[y]-v[x]!=w)flag=1;return ;}
	fa[fax]=fay;v[fax]=v[y]-v[x]-w;
}

int main()
{
	int T=read();
	register int i,x,y,w,n,m;
	while(T--)
	{
		n=read();m=read();
		flag=0;
		for(i=0;i<=n;++i)fa[i]=i;
		memset(v,0,sizeof(int)*(n+1));
		while(m--)
		{
			x=read()-1;y=read();w=read();
			judge(x,y,w);
		}
		flag?puts("false"):puts("true");
	}
	return 0;
}
/*
2
3 3
1 2 10
1 3 -5
3 3 -15
5 3
1 5 100
3 5 50
1 2 51

true
false
*/