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
思路:
带权并查集,存在权值为负数的情况,最后查询的时候,注意不能用 abs ,否则会出现错误 的答案。
#include <bits/stdc++.h>
#define mem(x,v) memset(x,v,sizeof(x))
#define go(i,a,b) for (int i = a; i <= b; i++)
#define og(i,a,b) for (int i = a; i >= b; i--)
#define MID(a,b) (a + b) / 2
#define lson now << 1
#define rson now << 1 | 1
using namespace std;
typedef long long LL;
const double EPS = 1e-10;
const int INF = 0x3f3f3f3f;
const int N = 1e3+10;
int f[N];
int val[N];
int n,m;
int find(int k){
if (f[k] == k) return k;
int temp = f[k];
f[k] = find(temp);
val[k] += val[temp];
return f[k];
}
int main(){
int T; cin>>T;
while(T--){
bool vis = 0;
scanf("%d%d",&n,&m);
go(i,0,n) f[i] = i;
mem(val,0);
go(i,1,m){
int x,y,xx,yy,z;
scanf("%d%d%d",&x,&y,&z);
x--;
xx = find(x); yy = find(y);
if (xx != yy){
f[xx] = yy;
val[xx] = val[y] + z - val[x];
}
if (xx == yy){
printf("%d %d\n",x,val[x]);
printf("%d %d\n",y,val[y]);
if ((val[x] - val[y]) != (z)) vis = 1;
}
}
if (vis) printf("false\n"); else
printf("true\n");
}
return 0;
}