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

方法：dp[i][j][0]表示在位置(i,j)上左边连续1的个数，dp[i][j][1]表示在位置(i,j)上上边连续1的个数

class Solution {
    public int largest1BorderedSquare(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;
        int[][][] dp = new int[m + 1][n + 1][2];

        for(int i = 1; i <= m; i++){
            for(int j = 1; j <= n; j++){
                if(grid[i - 1][j - 1] == 1){
                    dp[i][j][0] = dp[i][j - 1][0] + 1;
                    dp[i][j][1] = dp[i - 1][j][1] + 1;
                }
            }
        }
        int res = 0;
        for(int i = 1; i <= m; i++){
            for(int j = 1; j <= n; j++){
                for(int edge = Math.min(dp[i][j][0], dp[i][j][1]); edge >= 1; edge--){
                    if(dp[i][j - edge + 1][1] >= edge && dp[i - edge + 1][j][0] >= edge){
                        res = Math.max(res, edge);
                        break;
                    }
                }
            }
        }
        return res * res;
    }
}