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
}