持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第13天，点击查看活动详情
[面试题 17.23. 最大黑方阵](leetcode.cn/problems/sh…)
给定一个方阵，其中每个单元(像素)非黑即白。设计一个算法，找出 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
解题思路
计算每个位置元素向右和向下的最长连续黑色数目，保存在rm和dm数组；
遍历全部元素，对每一个查找它向右和向下最长连续黑色长度rm[r][c]和dm[r][c]中的较小值，作为遍历结尾e；
在[0,e]区间遍历t，对比rm[r+t][c]和dm[r][c+t]与t+1的大小，如果都非小于，则可以形成四边形，记录下来；
返回最终找到的最大结果。
代码实现
/**
* @param {number[][]} matrix
* @return {number[]}
*/
var findSquare = function(matrix) {
let h = matrix.length;
let w = matrix[0].length;
let rm = new Array(h).fill(0).map(i => new Array(w).fill(0));
let dm = new Array(h).fill(0).map(i => new Array(w).fill(0));
let max = [-1, -1, 0];
for (let r=0; r<h; r++) {
let cur = 0;
for (let c=w-1; c>=0; c--) {
if (matrix[r][c] === 0) {
cur += 1;
} else {
cur = 0;
}
rm[r][c] = cur;
}
}
for (let c=0; c<w; c++) {
let cur = 0;
for (let r=h-1; r>=0; r--) {
if (matrix[r][c] === 0) {
cur += 1;
} else {
cur = 0;
}
dm[r][c] = cur;
}
}
function find(r,c) {
let e = Math.min(rm[r][c], dm[r][c]);
let max = 0;
for (let t=0; t<e; t++) {
let dt = dm[r][c+t];
let rt = rm[r+t][c];
if (dt >= t+1 && rt >= t+1) max = Math.max(max, t+1);
}
return max;
}
for (let r=0; r<h; r++) {
for (let c=0; c<w; c++) {
let res = find(r,c);
if (res > max[2]) {
max[0] = r;
max[1] = c;
max[2] = res;
}
}
}
return max[0] === -1 ? [] : max;
};
如果你对这道题目还有疑问的话，可以在评论区进行留言；
