原题:
给你一个由若干 0 和 1 组成的二维网格 grid,请你找出边界全部由 1 组成的最大 正方形 子网格,并返回该子网格中的元素数量。如果不存在,则返回 0。
示例 1:
输入:grid = [[1,1,1],[1,0,1],[1,1,1]]
输出:9
示例 2:
输入:grid = [[1,1,0,0]]
输出:1
提示:
1 <= grid.length <= 100
1 <= grid[0].length <= 100
grid[i][j] 为 0 或 1
代码:
class Solution {
public:
int largest1BorderedSquare(vector<vector<int>>& grid)
{
vector<vector<int>> lefRig(grid.size(), vector<int>(grid[0].size(), 0));
vector<vector<int>> upDown(grid.size(), vector<int>(grid[0].size(), 0));
int ans = 0;
for (int i = grid.size() - 1; i >= 0; i--)
{
for (int j = 0; j<grid[i].size(); j++)
{
if(grid[i][j] == 1)
ans = 1;
if (i == grid.size() - 1)
{
upDown[i][j] = grid[i][j];
continue;
}
if (grid[i][j] != 0)
upDown[i][j] = upDown[i + 1][j] + 1;
else
upDown[i][j] = 0;
}
}
for (int i = 0; i<grid.size(); i++)
{
for (int j = 0; j<grid[i].size(); j++)
{
if (j == 0)
{
lefRig[i][j] = grid[i][j];
continue;
}
if (grid[i][j] == 1)
lefRig[i][j] = lefRig[i][j - 1] + 1;
else
lefRig[i][j] = 0;
}
}
for (int i = 0; i<grid.size(); i++)
{
for (int j = 0; j<grid[i].size(); j++)
{
// 下边界
int val = lefRig[i][j];
while (val >= 1 && val * val > ans)
{
if (val + i - 1 < grid.size() && lefRig[val + i - 1][j] >= val && upDown[i][j] >= val)
{
if (j - val + 1 >= 0 && upDown[i][j - val + 1] >= val)
{
ans = max(val * val, ans);
}
}
val--;
}
}
}
return ans;
}
};
解答:
如果做过航电上的1505和1506的两道题目,这道题应该顺理成章的会想出解决策略。
首先在横向和纵向计算每个位置可以得到的最大连续的宽度和最大连续高度。
例如有数据
1 1 0 1
1 1 1 0
1 0 1 0
1 1 1 1
得到的最大连续宽度lefRig和最大连续高度upDown的结果为
1 2 0 1
1 2 3 0
1 0 1 0
1 2 3 4
4 2 0 1
3 1 3 0
2 0 2 0
1 1 1 1
那么可以每次枚举lefRig[i][j]即每次枚举一个正方形上边的宽度,同时在当前枚举的坐标位置上,下移lefRig[i][j]个位置,查找lefRig表上查找下边对应的宽度,如下图第一个表的两个圈,值为3表示同时出现了宽度为3的边。
同理,在同样的位置上,在表upDown上寻找对应的高度,如下图第二个表两个圈。
在每次枚举lefRig表上的值时要注意,对应的下边宽度和对应的左右两边的高度可能会小于上边的宽度,由于要求时正方形,所以需要再走一个循环,每次减少当前枚举的上边的宽度,来寻找最大满足的正方形。