给你一个由若干 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
题解
每个点作为正方形的右下角的点进行比对,建立三维数组存放当前点上和左侧有多少个连续的1,再判断当前的点是否能构成正方形,然后更新最大值
代码
class Solution {
public:
int largest1BorderedSquare(vector<vector<int>>& grid) {
int res = 0;
int height = grid.size();
int width = grid[0].size();
vector<vector<vector<int>>> dp(height + 1, vector<vector<int>>(width + 1, vector<int>(2,0)));//dp[i][j][0] 代表一个点的向上的连续1的个数,dp[i][j][1]代表一个点的向左的连续1的个数。
for(int i = 1; i < height + 1; i++){
for(int j = 1; j < width + 1; j++){
if(grid[i - 1][j - 1] == 1){
dp[i][j][0] += dp[i - 1][j][0] + 1;
dp[i][j][1] += dp[i][j - 1][1] + 1;
}
int len = min(dp[i][j][0], dp[i][j][1]);
//int len = dp[i][j][0] > dp[i][j][1] ? dp[i][j][1] : dp[i][j][0];
while(len > res){
if(dp[i - len + 1][j][1] >= len && dp[i][j - len + 1][0] >= len){
res = max(res, len);
break;
}
len--;
}
}
}
return res * res;
}
};