leetcode之1139最大的以 1 为边界的正方形

给你一个由若干 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;
    }
};