leetcode221. 最大正方形

leetcode221. 最大正方形

五月 23, 2020

leetcode221. 最大正方形


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

示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4

动态规划

以左上角为顶点,能否构成正方形。
条件是

  • 当前matrix为1
  • dp的右侧,下面,右下角都有值
    具体在注释。
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    class Solution {
    public int maximalSquare(char[][] matrix) {
    //左上角为顶点
    if (matrix.length == 0) return 0;
    int row = matrix.length, co = matrix[0].length;
    if (co == 0) return 0;
    int[][] dp = new int[row][co];
    int max = 0;
    //初始化最右侧的列
    for (int i = row - 1; i >= 0; i--) {
    dp[i][co - 1] = (matrix[i][co - 1] == '1' ? 1 : 0);
    max = Math.max(max, dp[i][co - 1]);
    }
    //初始化最下面的行
    for (int i = co - 1; i >= 0; i--) {
    dp[row - 1][i] = (matrix[row - 1][i] == '1' ? 1 : 0);
    max = Math.max(max, dp[row - 1][i]);
    }
    //当前点能不能围成正方形,需要看我的右边,下边,右下角是什么情况,前提是当前点matrix为1
    /*
    matrix dp
    1 1 1 ? 2 1
    1 1 1 ->2 2 1
    1 1 1 1 1 1
    ?处填写需要满足三个数相等,或者取最小的,这样才能满足围成正方形的条件
    */
    //遍历从右往左,从下往上
    for (int i = row - 2; i >= 0; i--) {
    for (int j = co - 2; j >= 0; j--) {
    if (matrix[i][j] == '1') {
    dp[i][j] = 1;
    int right = dp[i][j + 1], down = dp[i + 1][j], rd = dp[i + 1][j + 1];
    int min = Math.min(Math.min(right, down), rd);
    if (min > 0) {
    dp[i][j] = 1 + min;
    }
    }
    max = Math.max(max, dp[i][j]);
    }
    }
    return max * max;
    }
    }
    leetcode 96/100