AT_arc090_b 题解

okmaker

已于 2025-06-25 17:35:05 修改

阅读量751

点赞数 23

CC 4.0 BY-SA版权

文章标签： c++ 数据结构算法

于 2025-06-25 17:34:27 首次发布

本文链接：https://blog.csdn.net/okmaker/article/details/148903601

使用带权并查集可以解决。

在维护并查集关系信息的同时，需要维护一个数组 $d i s$ 。
$d i s [i]$ 代表了从 $i$ 到当前族的根节点的距离。

问题1 如何维护dis信息

考虑如果有 $l$ 和 $r$ 两个节点，如果他们的根节点 $f l$ 和 $f r$ 并不是同一个，那么考虑合并。

我们将 $f a [f l] = f r$ ，也就是将 $l$ 连接到 $r$ ,这个时候我们需要维护 $d i s [f l]$ 的信息，也就是 $\rightarrow fr$ 的距离信息 $f l - f r$ 。

已知信息为 $r - l = d$ , $d i s [r] = r - f r$ , $d i s [l] = l - f l$ 。
那么 $d i s [r] - d i s [l] = f l - f r + r - l$ 。

这个时候移项可得: $d i s [r] - d i s [l] - (r - l) = f l - f r$ 。

也就是 $f l - f r = d i s [r] - d i s [l] - d$ 。得证。

问题2 如何查询

当两个节点为同一个组的时候，需要判断 $r - l == d$ 。
根据问题1的证明方式，易得： $d i s [r] - d i s [l] == r - l$ 。

code:

#include <bits/stdc++.h>
using namespace std;
#define close ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define int long long 
#define endl '\n'
int fa[MAXN],dis[MAXN];
//类似于后序遍历，一直维护当前节点到根节点的距离即可。
int find(int x){
    if(x != fa[x]){
        int t = fa[x];
        fa[x] = find(t);
        dis[x] += dis[t];
    }
    return fa[x];
}
signed main(){
    close
    int n,m; cin >> n >> m;
    for(int i = 1; i <= n; i++){
        fa[i] = i;
        dis[i] = 0;
    }
    bool ok = true;
    while(m--){
        int l,r,d; cin >> l >> r >> d;
        int fl = find(l),fr = find(r);
        if(fl == fr && dis[r] - dis[l] != d){
            ok = false;
        }
        if(fl != fr){
            // fl -> fr 
            fa[fl] = fr;
            // dis[r] 代表了 r到fr的距离 
            // dis[l] 代表了 l到lf的距离
            dis[fl] = dis[r] - dis[l] - d;
        }
    }
    if(ok) cout << "Yes" << endl;
    else cout << "No" << endl;
    return 0;
}