Hdu3038 How Mang Answers Wrong

题目链接: 点击跳转

题意：
题目会先输入n,m，分别表示有n个节点，m个回答，每个回答会告诉你节点l到节点r之间权值的总和(包含l，r)，当然，回答不一定是对的，需要我们根据输入顺序判断是否有错即可。

思路：
使用sum数组记录子节点到父节点的总和(l需要减一，因为l的权值也是包含在内的)，构建带权并查集，合并时注意因为两根节点合并会出现一条反边，即
sum[find(l)] = - sum[a] + sum[b] + v;所以这个权值是要被减去的，然后判断是否合理是看两个点在同一颗树上时，两者的差值与输入的权值是否相等，即v == sum[a] - sum[b] ? ‘Y’ : ‘N’；

代码如下:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 2e5 + 5e2;
int bin[MAXN], sum[MAXN];
int res = 0;
inline int find(int x) {
    if (x != bin[x]) {
        int temp = bin[x];
        bin[x] = find(bin[x]);
        sum[x] += sum[temp];
    }
    return bin[x];
}
inline void merge(int a, int b, int val) {
    int fa = find(a);
    int fb = find(b);
    if (fa == fb) {
        if (sum[a] - sum[b] != val) {
            res++;
        }
    } else {
        bin[fa] = fb;
        sum[fa] = - sum[a] + sum[b] + val;
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    int n, m;
    while (cin >> n >> m) {
        for (int i = 0; i <= n; i++) {
            bin[i] = i;
            sum[i] = 0;
        }
        res = 0;
        for (int i = 0; i < m; i++) {
            int l, r, v;
            cin >> l >> r >> v;l--;
            merge(l, r, v);
        }
        cout << res << endl;
    }
    return 0;
}