​LeetCode 547. 省份数量​

LeetCode 547. 省份数量

并查集裸题

问题的省份数量就是无向图中连通块的数量,考虑用并查集(Union Find)

并查集的原理:
我们只关心两个城市是否在同一个连通块(省份)内,而不关心它们的先后顺序,考虑在每个省份中选取一个城市作为省会,当判断两个城市是否在同一个省份时,只需要向前驱城市查询直到查到省会(前驱为自身),判断是否相等

初始化:
每个城市的前驱都为自己(即n个连通块)

查询:
查询前驱是否为省会,如果不是则继续递归查询前驱的前驱,直到查询到省会为止

合并:
查询两城市的省会
若在同一个省份则无需合并
否则将一个省会的前驱设为另一个省会,则这两个省份中的所有城市变为同一个省会,合并为一个省份

维护连通块(省份)个数:
两种思路:第一种是合并时维护,一开始省份数量为n,每进行一次有效合并都有两个省份变为一个,故省份数量减一
第二种是所有合并操作结束后,遍历每一个城市,查询省会的个数,即为省份的个数

我使用的是第二种

代码:
加入了路径压缩和按秩合并的优化

class Solution {
public:
    vector<int> f;
    vector<int> r;

    void init(int n) {
        f.resize(n + 1);
        r.resize(n + 1);
        for (int i = 0; i < n; i++)f[i] = i, r[i] = 0;
    }

    int find(int x) {
        return f[x] == x ? x : f[x] = find(f[x]);
    }

    bool combine(int x, int y) {
        int fx = find(x);
        int fy = find(y);
        if (fx == fy) return 0;
        if (r[fx] < r[fy]) swap(fx, fy);
        f[fy] = fx;
        r[fx] += r[fy];
        return 1;
    }

    int findCircleNum(vector<vector<int>>& isConnected) {
        int n = isConnected.size();
        init(n);
        for (int i = 0; i < n; i++)
            for (int j = i + 1; j < n; j++)
                if (isConnected[i][j])combine(i, j);

        int ans = 0;
        for (int i = 0; i < n; i++)
            if (i == f[i])ans++;

        return ans;
    }
};
  • 6
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值