[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;
}
- 可以用 \(d\) 数组维护节点到根节点的距离(也就是区间和),对于区间 \([l, r]\) ,和为 \(s\),合并 \(l - 1\) 和 \(r\) 所在集合,\(d[r] - d[l - 1]\) 即为区间 \([l, r]\) 的和
- 平行四边形法计算边权
\(d[pa] = d[b] + s - d[a]\),可以用下图所示的平行四边形计算:
- 对于题干中的和 \(s\),对应于上图中,应该是 \(b \rightarrow a = s\),即 \(a \rightarrow b = -s\)