【leetcode】1139 最大的以 1 为边界的正方形（数组）

题目链接：https://leetcode-cn.com/problems/largest-1-bordered-square/

题目描述

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

思路

/*
 * 建立预处理加速
 * 时间复杂度O(n^3) 空间复杂度O(n^2)
 */
class Solution {
public:
    int largest1BorderedSquare(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        vector<vector<int>> right(m, vector<int>(n, 0));    // right[i][j]表示从(i,j)出发向右连续1的个数
        vector<vector<int>> down(m, vector<int>(n, 0));     // down[i][j]表示从(i,j)出发向下连续1的个数
        initMap(grid,right,down);
        // 边长从大到小判断O(min(m,n))
        for (int len = min(m,n); len >0 ; --len) {
            if(hasLenOfSquare(len, right, down))
                return len*len;
        }
        return 0;
    }

private:
    // 判断是否有边长为len的正方形存在 O(n^2)
    bool hasLenOfSquare(int len, vector<vector<int>>&right, vector<vector<int>>& down){
        // 选取(i,j)作为左上角，判断是否存在边都是1的正方形
        for (int i = 0; i <= right.size()-len; ++i) {
            for (int j = 0; j <= right[0].size() - len ; ++j) {
                if(right[i][j]>= len && down[i][j] >= len
                    && down[i][j+len-1] >= len && right[i+len-1][j]>= len)
                    return true;
            }
        }
        return false;
    }
    // 建立记录表
    void initMap(vector<vector<int>> &m, vector<vector<int>> &right, vector<vector<int>> &down){
        int r = m.size()-1, c = m[0].size()-1;
        // 右下角元素
        if(m[r][c] == 1){
            right[r][c] = 1;
            down[r][c] = 1;
        }
        // 最后一列
        for (int i = r-1; i >= 0; --i) {
            if(m[i][c] == 1){
                right[i][c] = 1;
                down[i][c] = down[i+1][c] +1;
            }
        }
        // 最后一行
        for (int j = c-1; j >=0 ; --j) {
            if(m[r][j] == 1){
                right[r][j] = right[r][j+1] +1;
                down[r][j] = 1;
            }
        }

        // 其余位置元素
        for (int i = r-1; i >= 0; --i) {
            for (int j = c-1; j >=0 ; --j) {
                if(m[i][j] == 1){
                    right[i][j] = right[i][j+1] +1;
                    down[i][j] = down[i+1][j] +1;
                }
            }
        }
    }
};