[LeetCode]Maximal Square

Question

Given a 2D binary matrix filled with 0’s and 1’s, find the largest square containing only 1’s and return its area.

For example, given the following matrix:

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

Return 4.

---

本题难度Medium。

DP

复杂度

时间 O(MN) 空间 O(MN)

思路

我们知道，该点为右下角的正方形的最大边长，最多比它的上方，左方和左上方为右下角的正方形的边长多1，最好的情况是是它的上方，左方和左上方为右下角的正方形的大小都一样的，这样加上该点就可以构成一个更大的正方形。但如果它的上方，左方和左上方为右下角的正方形的大小不一样，合起来就会缺了某个角落，这时候只能取那三个正方形中最小的正方形的边长加1了。假设dp[i][j]表示以i,j为右下角的正方形的最大边长，则有：

dp[i][j] = min(dp[i-1][j-1], dp[i-1][j], dp[i][j-1]) + 1, if matrix[i][j]=’1’
dp[i][j] = 0, if matrix[i][j]=’0’

附

return value 是max的平方。

代码

public class Solution {
    public int maximalSquare(char[][] matrix) {
        //require
        int m = matrix.length;
        if(m<1)return 0;
        int n = matrix[0].length;
        int[][] dp = new int[m][n];
        int max = 0;
        //invariant
        //assign 0th row
        for(int i=0;i<n;i++){
            dp[0][i] = matrix[0][i]-'0';
            max = Math.max(max,dp[0][i]);
        }
        //assign 0th column
        for(int i=0;i<m;i++){
            dp[i][0] = matrix[i][0]-'0';
            max = Math.max(max,dp[i][0]);
        }
        //assign other elements
        for(int i=1;i<m;i++)
            for(int j=1;j<n;j++){
                dp[i][j] = matrix[i][j]!='1'?0:Math.min(Math.min(dp[i-1][j-1],dp[i][j-1]),dp[i-1][j])+1;
                max = Math.max(max,dp[i][j]);
            }
        //ensure
        return max*max;
    }
}

参考

[Leetcode] Maximal Square 最大正方形