并查集系列：547 Number of Provinces

Number of Provinces

There are n cities. Some of them are connected, while some are not. If city a is connected directly with city b, and city b is connected directly with city c, then city a is connected indirectly with city c.

A province is a group of directly or indirectly connected cities and no other cities outside of the group.

You are given an n x n matrix isConnected where isConnected[i][j] = 1 if the ith city and the jth city are directly connected, and isConnected[i][j] = 0 otherwise.

Return the total number of provinces.

Example 1:

Input: isConnected = [[1,1,0],[1,1,0],[0,0,1]]
Output: 2

Example 2:

Input: isConnected = [[1,0,0],[0,1,0],[0,0,1]]
Output: 3

Constraints:

• 1 <= n <= 200
• n == isConnected.length
• n == isConnected[i].length
• isConnected[i][j] is 1 or 0.
• isConnected[i][i] == 1
• isConnected[i][j] == isConnected[j][i]

并查集：

​ 这是一道明显到不能再明显的并查集的题目。

​ 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

​ 省份是一组直接或间接相连的城市，组内不含其他没有相连的城市。

​ 使用邻接矩阵来保存无向图。

​ 我的代码中用类实现了并查集的合并和查找操作，查找操作完成了路径压缩。

class DSU
{
    private:
        int n;
        vector<long> f;
    public:
        DSU(int N){
            n = N;
            f.resize(n, 1);
            for(int i=0; i<n; i++)
                f[i] = i;
        }
        int find(int x){
            if(x != f[x])
                f[x] = find(f[x]);
            return f[x];
        }
        void unionset(int x, int y){
            x = find(x);
            y = find(y);
            if(x == y)
                return;
            f[x] = y;
        }
};

class Solution {
public:
    int findCircleNum(vector<vector<int>>& isConnected) {
        int n = isConnected.size();
        DSU dsu(n);
        set<int> s;
        for(int i=0; i<n; i++){
            for(int j=i+1; j<n; j++)
                if(isConnected[i][j])
                    dsu.unionset(i, j);
        }
        for(int i=0; i<n; i++)
            s.insert(dsu.find(i));
        return s.size();
    }
};