[LeetCode] 417. 太平洋大西洋水流问题

一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第15天，点击查看活动详情。

417. 太平洋大西洋水流问题

有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 **“太平洋” **处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。

这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。

岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。

返回网格坐标 result 的 2D 列表 ，其中 result[i] = [r_i, c_i] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。

示例 1：

输入: heights = [[1,2,2,3,5],[3,2,3,4,4],[2,4,5,3,1],[6,7,1,4,5],[5,1,1,2,4]]
输出: [[0,4],[1,3],[1,4],[2,2],[3,0],[3,1],[4,0]]

示例 2：

输入: heights = [[2,1],[1,2]]
输出: [[0,0],[0,1],[1,0],[1,1]]

提示：

• m == heights.length
• n == heights[r].length
• 1 <= m, n <= 200
• 0 <= heights[r][c] <= 10⁵

BFS

func pacificAtlantic(heights [][]int) (ans [][]int) {
    m, n := len(heights), len(heights[0])
    bfs := func(queue []int) map[int]bool {
        res := map[int]bool{}
        for _, p := range queue {
            res[p] = true
        }
        for len(queue) > 0 {
            cur := queue[0]
            queue = queue[1:]
            x, y := cur / n, cur % n
            for _, dir := range [][]int{{0, 1}, {1, 0}, {0, -1}, {-1, 0}} {
                if nx, ny, nxt := x + dir[0], y + dir[1], (x + dir[0]) * n + (y + dir[1]); 0 <= nx && nx < m && 0 <= ny && ny < n && heights[nx][ny] >= heights[x][y] && !res[nxt] {
                    res[nxt] = true
                    queue = append(queue, nxt)
                }
            }
        }
        return res
    }

    pacific, atlantic := []int{}, []int{}
    for i := 0; i < n; i++ {
        pacific = append(pacific, i)
        atlantic = append(atlantic, (m - 1) * n + i)
    }
    for i := 1; i < m; i++ {
        pacific = append(pacific, i * n)
        atlantic = append(atlantic, (m - i) * n - 1)
    }
    m1, m2 := bfs(pacific), bfs(atlantic)
    for k := range m1 {
        if m2[k] {
            ans = append(ans, []int{k / n, k % n})
        }
    }
    return
}