我的leetcode答案都已经上传到我的githttps://github.com/ragezor/leetcode
题干
在一个由 0 和 1 组成的二维矩阵内,找到只包含 1 的最大正方形,并返回其面积。
示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4
想法
实际我们可以统计边长为0,1,2…的个数,然后取出最大的。
但是这是非常愚蠢的做法
于是想到动态规划
如果已知周围三个角的正方形的边长,
那么就可现在这个正方形的边长。
设dp[i][j]为以ij为右下角的正方形的边长
显然如果ij是‘1’,注意是‘1’
dp[i][j]=Math.min(Math.min(dp[i-1][j-1],dp[i-1][j]),dp[i][j-1])+1;
即另外三个方向的正方形的最小的大小加上自己现在的1。
那么代码就很好写了
看Java代码
Java代码
package daily;
public class MaximalSquare {
public int maximalSquare(char[][] matrix) {
int max = 0;
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return max;
}
int r = matrix.length, c = matrix[0].length;
int[][] dp = new int[r][c];
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
if (matrix[i][j] == '1') {
if (i == 0 || j == 0) {
dp[i][j] = 1;
} else {
dp[i][j] = Math.min(Math.min(dp[i - 1][j - 1], dp[i - 1][j]), dp[i][j - 1]) + 1;
max = Math.max(dp[i][j], max);
}
}
}
}
return max * max;
}
public static void main(String[] args) {
MaximalSquare maximalSquare = new MaximalSquare();
char[][] matrix = {{'1', '0', '1', '0', '0'},
{'1', '0', '1', '1', '1'},
{'1', '1', '1', '1', '1'},
{'1', '0', '0', '1', '0'}};
System.out.println(maximalSquare.maximalSquare(matrix));
}
}