LeeCode打卡第十八天：764最大加号标志

题目：

在一个 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 。

示例：

输入: n = 5, mines = [[4, 2]]
输出: 2
解释: 在上面的网格中，最大加号标志的阶只能是2。一个标志已在图中标出。

输入: n = 1, mines = [[0, 0]]
输出: 0
解释: 没有加号标志，返回 0 。

思路：

对于每个中心点坐标 (i,j)(i, j)(i,j)，分别从上下左右四个方向计算从 (i,j)(i, j)(i,j) 开始最长连续 111 的个数。设 dp[i][j][k]\textit{dp}[i][j][k]dp[i][j][k] 表示以 (i,j)(i, j)(i,j) 为起点在方向 kkk 上的连续 111 的最大数目：

如果 grid[i][j]\textit{grid}[i][j]grid[i][j] 为 000，那么此时该方向的连续 111 的最大数目为 000；
如果 grid[i][j]\textit{grid}[i][j]grid[i][j] 为 111, 那么此时该方向的连续 111 的最大数目为该方向上前一个单元为起点的最大数目加 111；
假设当前 k=0,1,2,3k = 0,1,2,3k=0,1,2,3 时，分别表示方向为左、右、上、下，则我们可以得到递推公式如下：

dp[i][j][0]={0,grid[i][j]=0dp[i][j−1][0]+1,grid[i][j]=1dp[i][j][1]={0,grid[i][j]=0dp[i][j+1][1]+1,grid[i][j]=1dp[i][j][2]={0,grid[i][j]=0dp[i−1][j][2]+1,grid[i][j]=1dp[i][j][3]={0,grid[i][j]=0dp[i+1][j][3]+1,grid[i][j]=1 \textit{dp}[i][j][0] = \begin{cases} 0, & \textit{grid}[i][j] = 0 \\ \textit{dp}[i][j-1][0] + 1 , & \textit{grid}[i][j] = 1 \\ \end{cases} \\ \textit{dp}[i][j][1] = \begin{cases} 0, & \textit{grid}[i][j] = 0 \\ \textit{dp}[i][j+1][1] + 1 , & \textit{grid}[i][j] = 1 \\ \end{cases} \\ \textit{dp}[i][j][2] = \begin{cases} 0, & \textit{grid}[i][j] = 0 \\ \textit{dp}[i-1][j][2] + 1 , & \textit{grid}[i][j] = 1 \\ \end{cases} \\ \textit{dp}[i][j][3] = \begin{cases} 0, & \textit{grid}[i][j] = 0 \\ \textit{dp}[i+1][j][3] + 1 , & \textit{grid}[i][j] = 1 \\ \end{cases} \\
dp[i][j][0]={ 
0,
dp[i][j−1][0]+1,
​
  
grid[i][j]=0
grid[i][j]=1
​
 
dp[i][j][1]={ 
0,
dp[i][j+1][1]+1,
​
  
grid[i][j]=0
grid[i][j]=1
​
 
dp[i][j][2]={ 
0,
dp[i−1][j][2]+1,
​
  
grid[i][j]=0
grid[i][j]=1
​
 
dp[i][j][3]={ 
0,
dp[i+1][j][3]+1,
​
  
grid[i][j]=0
grid[i][j]=1
​
 
假设网格中有一行为 01110110\texttt{01110110}01110110，当前方向为向左，那么对应的连续 111 的个数就是 012301201012301201012301201。以每个点为 (i,j)(i,j)(i,j) 为中心的四个方向中最小连续 111 的个数即为以其为中心构成的加号标志的最大阶数，我们用公式表示 L=min⁡k=03dp[i][j][k]L = \min\limits_{k=0}^{3}\textit{dp}[i][j][k]L= 
k=0
min
3
​
 dp[i][j][k]。在实际计算时，我们为了方便计算只用 dp[i][j]\textit{dp}[i][j]dp[i][j] 保存四个方向中最小的连续 111 的个数即可。

代码：

class Solution:

    def orderOfLargestPlusSign(self, n: int, mines: List[List[int]]) -> int:

        dp = [[n] * n for _ in range(n)]

        banned = set(map(tuple, mines))

        for i in range(n):

            # left

            count = 0

            for j in range(n):

                count = 0 if (i, j) in banned else count + 1

                dp[i][j] = min(dp[i][j], count)

            # right

            count = 0

            for j in range(n - 1, -1, -1):

                count = 0 if (i, j) in banned else count + 1

                dp[i][j] = min(dp[i][j], count)

        for j in range(n):

            # up

            count = 0

            for i in range(n):

                count = 0 if (i, j) in banned else count + 1

                dp[i][j] = min(dp[i][j], count)

            # down

            count = 0

            for i in range(n - 1, -1, -1):

                count = 0 if (i, j) in banned else count + 1

                dp[i][j] = min(dp[i][j], count)

        return max(map(max, dp))