Leetcode: NO.221 最大正方形

题目

题目链接：https://leetcode-cn.com/problems/maximal-square

在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。

示例:

输入: 

1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

输出: 4

解题记录

• 遍历求解，如果是1的话，判断其右，下，右下是否都为1，如果都为1的话说明满足面积为4的正方形，否则记录面积为1
• 满足面积为4，需要进一步判断是否满足更大的面积，通过边界计算出可能最大的边长，逐个判断
• 获取最大面积即可

class Solution {
    public static int maximalSquare (char[][] matrix) {
        int maxY = matrix.length;
        if (maxY==0) return 0;
        int maxX = matrix[0].length;
        int area = 0;
        for (int y=0; y<maxY; ++y){
            for (int x=0; x<maxX; ++x) {
                if (matrix[y][x]=='1'){
                    if (x!=maxX-1&&y!=maxY-1&&matrix[y][x+1]=='1'&&matrix[y+1][x]=='1'&&matrix[y+1][x+1]=='1'){
                        area = Math.max(area, 4);
                        int maxLine = Math.min(maxX-x, maxY-y);
                        if (maxLine>2) {
                            area = Math.max(area, checkSquare(x,y,maxLine,matrix));
                        }
                    }
                    area = Math.max(area, 1);
                }
            }
        }
        return area;
    }

    static int checkSquare (int x, int y, int maxLine, char[][] matrix) {
        for (int i=2; i<maxLine; ++i){
            for (int j=0; j<=i; ++j){
                if (matrix[y+i][x+j]=='0'|| matrix[y+j][x+i]=='0') return i*i;
            }
        }
        return maxLine*maxLine;
    }
}

进阶

使用动态规划求解

• 这里我们通过右下角向左上查找，看是否存在正方形
• 通过一个dp矩阵记录该点到0的距离
• 值为0的话距离自然为0，值为1的话距离为1
• 右下角的值为，上，左，坐上中距离0最小值再加上1
  
• 状态方程为 dp(i, j) = min(dp(i-1, j), dp(i, j-1), dp(i-1, j-1)) + 1

class Solution {
    public int maximalSquare (char[][] matrix) {
        int maxY = matrix.length;
        if(maxY==0) return 0;
        int maxX = matrix[0].length;
        int line = 0;
        int[][] dp = new int[maxY][maxX];
        for (int y=0; y<maxY; ++y){
            for (int x=0; x<maxX; ++x){
                if (matrix[y][x]=='1'){
                    dp[y][x]=1;
                    line=1;
                }        
            }
        }

        for (int y=1; y<maxY; ++y){
            for (int x=1; x<maxX; ++x){
                if (matrix[y][x] == '1'){
                    dp[y][x] = Math.min(dp[y-1][x-1], Math.min(dp[y-1][x], dp[y][x-1])) + 1;
                    line = Math.max(dp[y][x], line);
                }
            }
        }
        return line*line;
    }
}