难度中等12
给定一个方阵,其中每个单元(像素)非黑即白。设计一个算法,找出 4 条边皆为黑色像素的最大子方阵。
返回一个数组 [r, c, size]
,其中 r
, c
分别代表子方阵左上角的行号和列号,size
是子方阵的边长。若有多个满足条件的子方阵,返回 r
最小的,若 r
相同,返回 c
最小的子方阵。若无满足条件的子方阵,返回空数组。
示例 1:
输入:
[
[1,0,1],
[0,0,1],
[0,0,1]
]
输出: [1,0,2]
解释: 输入中 0 代表黑色,1 代表白色,标粗的元素即为满足条件的最大子方阵
示例 2:
输入:
[
[0,1,1],
[1,0,1],
[1,1,0]
]
输出: [0,0,1]
提示:
matrix.length == matrix[0].length <= 200
对整个图进行依次遍历,通过DP来更新每个点对应的向上和向左的连续黑色长度,进而判断其上顶点和左顶点所对应的DP[]值是否满足正方形条件,再通过比较Max来更新结果。
下面的代码是可以继续优化的,在正方形判断之前可以比较当前找到的最大长度与该点的可能长度(大于就无需继续判断),