acwing238. 银河英雄传说(带权并查集)

题意

在这里插入图片描述

思路

  1. 这题明显的是带权并查集,但是我们要多维护一个集合大小 num [] 数组,因为我们这题的带权并查集维护的不仅是节点之间的相互关系,而需要维护的是实际真实的距离。有 num 之后就方便我们维护实际距离了。

  2. 带权并查集核心
    在这里插入图片描述

  3. 维护真实边权的核心:
    在这里插入图片描述

代码

/* #include <iostream> */
#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 30005;
int f[N], d[N], num[N];

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


int main()
{
    
    
    for (int i = 1; i < N; i ++) f[i] = i, num[i] = 1, d[i] = 0;
    int m; scanf("%d ", &m);
    char ch; int x, y;
    while (m --)
    {
    
    
        scanf("%c %d %d ", &ch, &x, &y);
        int fx = find(x), fy = find(y);
        if (ch == 'M')
        {
    
    
            f[fx] = fy;
            d[fx] = num[fy];
            num[fy] += num[fx];
        }
        else
        {
    
    
            int dis = -1;
            if (fx == fy)
                dis = max(abs(d[x] - d[y]) - 1, 0);
            printf("%d\n", dis);
        }
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_34261446/article/details/121159141