时间复杂度:O(n²),空间复杂度:O(n²)
解题思路
深度优先搜索。
从岛屿的一个点开始向四周探索,并维护一个visit二维数组保存遍历过的位置。遍历完一个岛屿后会重新选择另一个没有遍历的岛屿,再次重复以上步骤。直至遍历完全部元素。
AC代码
func numIslands(grid [][]byte) (res int) {
m,n:=len(grid),len(grid[0])
//创建指定长度的二维切片
visit:=make([][]bool,m)
for i:=0;i<m;i++{
visit[i]=make([]bool,n)
}
for i:=0;i<m;i++{
for j:=0;j<n;j++{
if grid[i][j]=='1'&&!visit[i][j]{
DFS(grid,&visit,i,j)
res++
}
}
}
return res
}
func DFS(grid [][]byte,visit *[][]bool,x,y int){
(*visit)[x][y]=true
//判断上方
if x-1>=0&&!(*visit)[x-1][y]&&grid[x-1][y]=='1'{
DFS(grid,visit,x-1,y)
}
//判断右侧
if y+1<len(grid[0])&&!(*visit)[x][y+1]&&grid[x][y+1]=='1'{
DFS(grid,visit,x,y+1)
}
//判断下方
if x+1<len(grid)&&!(*visit)[x+1][y]&&grid[x+1][y]=='1'{
DFS(grid,visit,x+1,y)
}
//判断左侧
if y-1>=0&&!(*visit)[x][y-1]&&grid[x][y-1]=='1'{
DFS(grid,visit,x,y-1)
}
}
感悟
再一次复习了用go语言创建指定长度的二维切片,巩固了go语言基础。
第二是DFS的visit参数是引用传递而不是值传递,原因是我们需要在DFS中修改visit切片,如果是普通的值传递将无法修改外部真正的visit切片,修改的是它在DFS中的一个副本。
另外grid切片也可以使用引用传递,这样不会单独创建一个副本,性能会略微提升,不过代码写起来会更麻烦一些,需要将grid解引用。