[AcWing 4286] 多少个答案是错误的

带权并查集

---

点击查看代码

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;

const int N = 1e6 + 10;

int n, m;
int p[N], d[N];

int find(int x)
{
    if (p[x] != x) {
        int root = find(p[x]);
        d[x] += d[p[x]];
        p[x] = root;
    }
    return p[x];
}

void merge(int a, int b, int s)
{
    int pa = find(a), pb = find(b);
    d[pa] = d[b] + s - d[a];
    p[pa] = pb;
}

void solve()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++)
        p[i] = i;
    int res = 0;
    while (m --) {
        int a, b, s;
        cin >> a >> b >> s;
        a --;
        if (find(a) == find(b)) {
            res += (d[b] - d[a] != s);
        }
        else
            merge(a, b, -s);
    }
    cout << res << endl;
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    solve();

    return 0;
}

---

1. 可以用 $d$ 数组维护节点到根节点的距离（也就是区间和），对于区间 $[l, r]$ ，和为 $s$，合并 $l - 1$ 和 $r$ 所在集合，$d[r] - d[l - 1]$ 即为区间 $[l, r]$ 的和
2. 平行四边形法计算边权
   $d[pa] = d[b] + s - d[a]$，可以用下图所示的平行四边形计算：
   
3. 对于题干中的和 $s$，对应于上图中，应该是 $b \rightarrow a = s$，即 $a \rightarrow b = -s$