【Leetcode】1765. Map of Highest Peak

题目地址：

https://leetcode.com/problems/map-of-highest-peak/

给定一个$m\times n$的$0-1$矩阵$A$，要求构造一个新的矩阵$C$满足如下要求：
1、$C$的所有元素都非负；
2、若$A[i][j]=1$则$C[i][j]=0$；
3、对于$C$中相邻的元素，它们的差的绝对值最多是$1$。这里的相邻指的是四相邻。
同时要求$C$的最大值最大，求$C$。

其实只需要从$C[i][j]=0$的位置做BFS即可，第一次接触的位置为它的前驱的数加$1$。我们稍微说明一下正确性。首先如果某两个相邻格子元素之差大于$1$了，比如$C[x_1][y_1]>C[x_2][y_2]+1$那么说明$(x_1,y_1)$这个位置应该由$(x_2,y_2)$这个位置扩展而来（因为$(x_2,y_2)$入队之后下一层BFS就会拓展到$(x_1,y_1)$），这就矛盾了，因为我们不允许同一个位置被BFS到两次。至于最大值最大这一点，可以反证法，如果某个位置的能达到更大值，那么考虑BFS树从树根走到这个位置的路径（可以想象树根是个虚拟节点），这个路径必然存在相邻两个位置差大于$1$，矛盾。代码如下：

import java.util.LinkedList;
import java.util.Queue;

public class Solution {
    public int[][] highestPeak(int[][] A) {
        int m = A.length, n = A[0].length;
        Queue<int[]> queue = new LinkedList<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (A[i][j] == 1) {
                    A[i][j] = 0;
                    queue.offer(new int[]{i, j});
                } else {
                    A[i][j] = -1;
                }
            }
        }
        
        int[] d = {1, 0, -1, 0, 1};
        
        for (int h = 1; !queue.isEmpty(); h++) {
            for (int i = queue.size() - 1; i >= 0; i--) {
                int[] cur = queue.poll();
                int x = cur[0], y = cur[1];
                for (int k = 0; k < 4; k++) {
                    int nextX = x + d[k], nextY = y + d[k + 1];
                    if (0 <= nextX && nextX < m && 0 <= nextY && nextY < n && A[nextX][nextY] == -1) {
                        A[nextX][nextY] = h;
                        queue.offer(new int[]{nextX, nextY});
                    }
                }
            }
        }
        
        return A;
    }
}

时空复杂度$O(mn)$。