LeetCode刷题：547. 省份数量（golang版）

• 题目描述

有 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-provinces
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例一图：

示例二图：

• 思路
  1、可以使用并查集，初始时，每个城市都属于不同的连通分量，其父节点是它自己。遍历矩阵isConnected，如果两个城市之间有相连关系，则它们属于同一个连通分量，对它们进行合并
  2、遍历矩阵isConnected 的全部元素之后，计算连通分量的总数，即为省份的总数。
• 代码

package main

import "fmt"

func findCircleNum(isConnected [][]int) (ans int) {
	n := len(isConnected)
	parent := make([]int,n)
	//初始化其父节点，让每个节点的父节点等于自己
	for k := range parent {
		parent[k] = k
	}
	//遍历isConnected
	for i:=0; i<n; i++ {
		//因为是对称矩阵所以j可以从i+1开始遍历
		for j:=i+1; j<n; j++ {
			if isConnected[i][j]==1 {
				//找到j节点的父节点，将i节点的父节点的值赋值给j，连通他们
				parent[findParent(&parent,j)] = findParent(&parent,i)
			}
		}
	}
	//返回结果
	ans = 0
	for k := range parent {
		if parent[k] == k {
			ans ++
		}
	}
	return ans
}
//递归查找一个节点的父节点，直到节点的父节点是自己为止
func findParent(parent *[]int,x int) int {
	if (*parent)[x] != x {
		(*parent)[x] = findParent(parent, (*parent)[x])
	}
	return (*parent)[x]
}

func main() {
	var grid = make([][]int, 4)
	grid[0] = []int{1,0,0,1}
	grid[1] = []int{0,1,1,0}
	grid[2] = []int{0,1,1,1}
	grid[3] = []int{1,0,1,1}

	res := findCircleNum(grid)
	fmt.Println(res)
}