LeetCode第一题——曼哈顿距离

你现在手里有一份大小为 N x N 的『地图』(网格) grid,上面的每个『区域』(单元格)都用 0 和 1 标记好了。其中 0 代表海洋,1 代表陆地,你知道距离陆地区域最远的海洋区域是是哪一个吗?请返回该海洋区域到离它最近的陆地区域的距离。

我们这里说的距离是『曼哈顿距离』( Manhattan Distance):(x0, y0) 和 (x1, y1) 这两个区域之间的距离是 |x0 - x1| + |y0 - y1| 。

如果我们的地图上只有陆地或者海洋,请返回 -1。

示例 1:

1 0 1
0 0 0
1 0 1

输入:[[1,0,1],[0,0,0],[1,0,1]]
输出:2
解释:
海洋区域 (1, 1) 和所有陆地区域之间的距离都达到最大,最大距离为 2。

class Solution {
    public int maxDistance(int[][] grid) {
        int N = grid.length;
        
        Queue<int[]> queue = new ArrayDeque<>();
        // 将所有的陆地格子加入队列
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (grid[i][j] == 1) {
                    queue.add(new int[]{i, j});
                }
            }
        }
        // 如果我们的地图上只有陆地或者海洋,请返回 -1。
        if (queue.isEmpty() || queue.size() == N * N) {
            return -1;
        }
        
        int distance = -1;
        while (!queue.isEmpty()) {
            distance++;
            int n = queue.size();
            // 这里一口气取出 n 个结点,以实现层序遍历
            for (int i = 0; i < n; i++) {
                int[] cell = queue.poll();     //poll()移除并返问队列头部的元素 
                int r = cell[0];
                int c = cell[1];
                // 遍历上方单元格
                if (r-1 >= 0 && grid[r-1][c] == 0) {  //cell点正上方存在0
                    grid[r-1][c] = 2;
                    queue.add(new int[]{r-1, c});
                }
                // 遍历下方单元格
                if (r+1 < N && grid[r+1][c] == 0) {	   //cell点正下方存在0
                    grid[r+1][c] = 2;
                    queue.add(new int[]{r+1, c});
                }
                // 遍历左边单元格
                if (c-1 >= 0 && grid[r][c-1] == 0) {   //cell点正左边存在0
                    grid[r][c-1] = 2;
                    queue.add(new int[]{r, c-1});
                }
                // 遍历右边单元格
                if (c+1 < N && grid[r][c+1] == 0) {    //cell点正右边存在0
                    grid[r][c+1] = 2;
                    queue.add(new int[]{r, c+1});
                }
            }
        }  
        return distance;

}

思路:首先把所有的陆地放在一个队列中,然后执行第一次循环,将与原陆地距离为1的海洋写入改队列中,然后把原陆地全部删除;然后执行第二次循环,将于原陆地距离为2的海洋写入该队列中,然后删除与原陆地距离为1的所有海洋;以此类推,当队列中没有海洋的时候,就结束循环。所以最远距离就是当队列中没有值时的距离-1,所以为了方便我们把起始距离设为-1,则最后返回的就是最远的距离。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章