LeetCode-547. 省份数量 - sheep44044的小站

图论#

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

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

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中省份的数量。

示例 1：

1
输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
2
输出：2

示例 2：

1
输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
2
输出：3

并查集的方法

感觉是一个很神奇的方法

1
func findCircleNum(isConnected [][]int) int {
2
    n := len(isConnected)
3

4
    parent := make([]int, n)
5
    for i := 0; i < n; i++ {
6
        parent[i] = i
7
    }
8

9
    var find func(i int) int
10
    find = func(i int) int{
11
        if parent[i] != i {
12
            parent[i] = find(parent[i])
13
        }
14
        return parent[i]
15
    }
16

17
    count := n
18
    for i := 0; i < n; i++ {
19
        for j := i+1; j < n; j++ {
20
            if isConnected[i][j] == 1 {
21
                rootI := find(i)
22
                rootJ := find(j)
23
                if rootI != rootJ {
24
                    parent[rootI] = rootJ
25
                    count--
26
                }
27
            }
28
        }
29
    }
30
    return count
31
}