[HNOI2005]狡猾的商人

题目描述

输入输出格式

输入格式：

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

输出格式：

输出文件output.txt中包含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

 题解：

这个题目是一个带权并查集的裸题，记sum[i]表示i到根的前缀合之差，自己推一下式子就可以了，注意，因为是前缀和，所以前面一个要减去1，并且find的时候要记下father，先更改值，再更新sum。

代码：

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<cstring>
using namespace std;
int fa[10000],sum[10000];
int n,m,w,flag;

void cl(){
    memset(sum,0,sizeof(sum));
    memset(fa,0,sizeof(fa));
    for(int i=1;i<=n;i++) fa[i]=i;
}

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

void hebin(int x,int y,int quan){
    int roofx=find(x),roofy=find(y);
    if(roofx>roofy) sum[roofx]=sum[y]-sum[x]-quan,fa[roofx]=roofy;
    else sum[roofy]=sum[x]-sum[y]+quan,fa[roofy]=roofx;
}

int main()
{
    scanf("%d",&w);
    while(w--){
        scanf("%d%d",&n,&m);
        cl();flag=1;
        for(int i=1;i<=m;i++){
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            if(!flag) continue;
            x--;
            if(find(x)!=find(y)) 
                hebin(x,y,z);
            else if(sum[y]-sum[x]!=z) {flag=0;}
         }
        if(flag) printf("true
");
        else printf("false
");
    }
}