leetcode 探索 队列与栈 岛屿数量

题目

给你一个由 ‘1’（陆地）和 ‘0’（水）组成的的二维网格，请你计算网格中岛屿的数量。
岛屿总是被水包围，并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。
此外，你可以假设该网格的四条边均被水包围。
示例 1:

输入:
11110
11010
11000
00000
输出: 1

示例 2:

输入:
11000
11000
00100
00011
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-islands

分析

就是探索连城一片的1的个数，采用广度优先搜索结合队列实现，Go中没有队列，采用slice实现，每次出队slice就重新赋值。你可以把注释掉的打印语句打开，方便理解。

解法

func numIslands(grid [][]byte) int {
    if len(grid) == 0{
        return 0
    }
    nr := len(grid)
    nc := len(grid[0])
    
    var numIsland = 0
    for r := range grid {
        // fmt.Printf("r: %d\n", r)
        for c := range grid[0] {
            // fmt.Printf("c: %d\n", c)
            
            if grid[r][c] == '1' {
                numIsland++
                grid[r][c] = '0'

                neighbors := make([]pair, 0)
                neighbors = append(neighbors, pair{r, c})
                for len(neighbors) != 0 {
                    front := neighbors[0]   // front
                    neighbors = neighbors[1:] // dequeue
                    row, col := front.first, front.second
                    // fmt.Printf("front: %d, %d\n", row,col)
                    if row - 1 >= 0 && grid[row-1][col] == '1' {
                        neighbors = append(neighbors, pair{row-1, col}) // enqueue
                        grid[row - 1][col] = '0' // 标记
                    }
                    if row + 1 < nr && grid[row+1][col] == '1' {
                        neighbors = append(neighbors, pair{row+1, col}) // enqueue
                        grid[row+1][col] = '0' // 标记
                    }
                    if col + 1 < nc && grid[row][col+1] == '1' {
                        neighbors = append(neighbors, pair{row, col+1}) // enqueue
                        grid[row][col+1] = '0' // 标记
                    }
                    if col - 1 >= 0 && grid[row][col-1] == '1' {
                        neighbors = append(neighbors, pair{row, col-1}) // enqueue
                        grid[row][col-1] = '0' // 标记
                    }
                    
                    // fmt.Printf("queue is: %v\n", neighbors)
                }
            }            
        }
    }
    
    return numIsland
}

type pair struct {
    first int
    second int
}