leetcode:最大加号标志

一、题目

在一个 n x n 的矩阵 grid 中，除了在数组 mines 中给出的元素为 0，其他每个元素都为 1。mines[i] = [xi, yi]表示 grid[xi][yi] == 0

返回  grid 中包含 1 的最大的 轴对齐 加号标志的阶数 。如果未找到加号标志，则返回 0 。

一个 k 阶由 1 组成的 “轴对称”加号标志 具有中心网格 grid[r][c] == 1 ，以及4个从中心向上、向下、向左、向右延伸，长度为 k-1，由 1 组成的臂。注意，只有加号标志的所有网格要求为 1 ，别的网格可能为 0 也可能为 1 。

来源：力扣

二、思路

我们定义 dp[i][j]dp[i][j] 表示以 (i, j)(i,j) 为中心的最大加号标志的阶数，答案即为所有 dp[i][j]dp[i][j] 的最大值。

我们可以发现，对于每个 (i, j)(i,j)，其最大加号标志的阶数不会超过其上下左右四个方向上连续的 11 的个数的最小值。因此，我们可以预处理出每个位置上下左右四个方向上连续的 11 的个数，然后遍历所有的 (i, j)(i,j)，求出 dp[i][j]dp[i][j] 的最大值即可。
 

三、代码

class Solution:
    def orderOfLargestPlusSign(self, n: int, mines: List[List[int]]) -> int:
        dp = [[n] * n for _ in range(n)]
        for x, y in mines:
            dp[x][y] = 0
        for i in range(n):
            left = right = up = down = 0
            for j, k in zip(range(n), reversed(range(n))):
                left = left + 1 if dp[i][j] else 0
                right = right + 1 if dp[i][k] else 0
                up = up + 1 if dp[j][i] else 0
                down = down + 1 if dp[k][i] else 0
                dp[i][j] = min(dp[i][j], left)
                dp[i][k] = min(dp[i][k], right)
                dp[j][i] = min(dp[j][i], up)
                dp[k][i] = min(dp[k][i], down)
        return max(max(v) for v in dp)