剑指 Offer II 116. 省份数量

省份数量

题目描述:

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

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

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

返回矩阵中 省份 的数量。

示例 1:

输入:isConnected = [[1,1,0],[1,1,0],[0,0,1]]
输出:2
示例 2:

输入:isConnected = [[1,0,0],[0,1,0],[0,0,1]]
输出:3

提示:

1 <= n <= 200
n == isConnected.length
n == isConnected[i].length
isConnected[i][j] 为 1 或 0
isConnected[i][i] == 1
isConnected[i][j] == isConnected[j][i]

思路:

用并查集思路,两个城市连起来就把city【i】 =j,判断city【i】是否等于i就知道有几个省了,并查集合并有点困难

代码:

func findCircleNum(isConnected [][]int) int {
	var city [201]int
	var res int
	for i := 0; i < 201; i++ {
		city[i] = i
	}
	var find func(int) int
	find = func(x int) int {
		if city[x] != x {
			city[x] = find(city[x])
		}
		return city[x]
	}
	for i := 1; i < len(isConnected); i++ {
		for j := 0; j < i; j++ {
			if isConnected[i][j] == 1 {
				city[find(i)] = find(j)
			}
		}
	}
	for i := 0; i < len(isConnected); i++ {
		if city[i] == i {
			res++
		}
	}
	return res
}

代码效率:

执行用时:24 ms, 在所有 Go 提交中击败了91.90%的用户
内存消耗:6.5 MB, 在所有 Go 提交中击败了95.76%的用户