【BZOJ】1202: [HNOI2005]狡猾的商人（并查集+前缀和）

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

用并查集+前缀和。

前缀和从后向前维护和，并查集从前往后合并

对于询问l, r

如果l-1和r是一个集合（在这里，并查集每个集合都是一个可行的区间），那么直接判断s[l-1]-s[r]是否等于所给值

否则要合并l-1和r的集合（画图可知，p[fx]=fy时，s[fx]=s[y]-s[x]+所给值，这是由并查集的实现决定的）

而在并查集里维护前缀和就是用s[x]+=s[fa]，而在并查集中，x会被合并到一个新的集合的根中，所以不会重复累计x的fa。

#include <cstdio>

#include <cstring>

#include <string>

#include <iostream>

#include <cmath>

#include <algorithm>

using namespace std;

#define for1(i,a,n) for(i=a;i<=n;++i)

#define for2(i,a,n) for(i=a;i<n;++i)

#define for3(i,a,n) for(i=a;i>=n;--i)

#define for4(i,a,n) for(i=a;i>n;--i)

#define CC(i,a) memset(i,a,sizeof(i))

#define max(a,b) ((a)>(b)?(a):(b))

#define min(a,b) ((a)<(b)?(a):(b))

#define read(a) a=getnum()

#define print(a) printf("%d", a)

inline int getnum() { int ret=0; char c; int k=1; for(c=getchar(); c<'0' || c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0' && c<='9'; c=getchar()) ret=ret*10+c-'0'; return k*ret; }

const int N=1005;

int s[N], p[N];

int ifind(const int &x) {

	if(x==p[x]) return x;

	int fa=p[x]; p[x]=ifind(p[x]);

	s[x]+=s[fa];

	return p[x];

}

int main() {

	int n, m, w, i, t, x, y, fx, fy, ok;

	read(w);

	while(w--) {

		read(n); read(m);

		ok=1;

		CC(s, 0);

		for1(i, 0, n+2) p[i]=i;

		for1(i, 1, m) {

			read(x); read(y); read(t); --x;

			fx=ifind(x); fy=ifind(y);

			if(fx!=fy) {

				p[fx]=fy;

				s[fx]=s[y]-s[x]+t;

			}

			else if(s[x]-s[y]!=t) { ok=0; break; }

		}

		ok?puts("true"):puts("false");

	}

	return 0;

}

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