Hdu3038 How Mang Answers Wrong

本文最后更新于:3 years ago

题目链接: 点击跳转

题意:
题目会先输入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’;

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#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; //将fa接到fb下,即a与da的边作为反边减掉
}
}
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;
}

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!