当前位置:   article > 正文

算法49:动态规划专练(力扣1139题:最大正方形面积)

算法49:动态规划专练(力扣1139题:最大正方形面积)

题目:

给你一个由若干 0 和 1 组成的二维网格 grid,请你找出边界全部由 1 组成的最大 正方形 子网格,并返回该子网格中的元素数量。如果不存在,则返回 0

示例 1:

输入:grid = [[1,1,1],[1,0,1],[1,1,1]]
输出:9

示例 2:

输入:grid = [[1,1,0,0]]
输出:1

这一题与算法48有点相似,但是它不能用单调栈解决这个问题。也不能用算法48的动态规划思想来解决。因为这一题是以全部为1的边组成的最大面积。也就是说中间是可以为0的。式例1就是最好的说明。

上一题用了单调栈、动态规划、暴力解。而这一题已经把前两种可能性已经排除掉了。只剩下暴力解了。那么,暴力解如何来解决这一题呢?

1. 正方形有4个顶点、4条边。只要验证每条边是否全部为1就可以了

2. 每增加一行一列,就验证新增的这些边是否全部为1就可以。如果验证不通过,没关系,继续往后验证。因为,当前新增的边并不一定就是最终的正方形边长,如果它只是最大正方形内部的一些元素而已,那我们根本不关注它是否为0.   

3. 需要注意的是,正方形的上方边和左方边出现了0, 那就不能继续去判断了。因为一点正方形的左上方顶点一旦确定,那上方边长和左方边长就已经确定了延长的方向了。但是,右侧的边和下方的边是要全部遍历完,才能最后确定的。

  1. package code04.动态规划专项训练02;
  2. /**
  3. * 力扣 1139 最大的以1为边界的正方形
  4. * https://leetcode.com/problems/largest-1-bordered-square/
  5. */
  6. public class Largest1BorderedSquare_04_leetcode1139暴力解 {
  7. public int largest1BorderedSquare( int[][] matrix) {
  8. if (matrix == null || matrix.length == 0) {
  9. return 0;
  10. }
  11. int row = matrix.length;
  12. int col = matrix[0].length;
  13. //全局正方形最大边长
  14. int maxSide = 0;
  15. for (int i = 0; i < row ; i++) {
  16. //当前行的每一列都作为正方形的左上角,即起始点
  17. for (int j = 0; j < col; j++) {
  18. //当前格子是否为1,为1才可能成为正方形的起始点
  19. if (matrix[i][j] == 0) {
  20. continue;
  21. }
  22. //整个二维数组中,重要有1出现,那正方形边长至少为1
  23. if (maxSide == 0) {
  24. maxSide = 1;
  25. }
  26. //行边长,列边长,两者取小。因为是正方形
  27. int p = Math.min(row - i, col - j);
  28. //当前单元格 matrix[i][j] 开始的最大正方形边长
  29. int tempMaxSide = 1;
  30. boolean flag = true;
  31. //从i,j开始 到 i+index,j+index 范围内。全部都为1,才能验证通过
  32. //start为默认边长,默认边长为1. 因为matrix[i][j] == 1
  33. for (int count = 1; count < p; count++) {
  34. //上方边长 matrix[i][j+count] == '0'
  35. //左方边长 matrix[i + count][j] == '0'
  36. if (matrix[i][j+count] == 0
  37. || matrix[i + count][j] == 0) {
  38. break;
  39. }
  40. int addCol = j + count;
  41. int addRow = i + count;
  42. //新增的行和列,不是全部范围
  43. for (int m = 1; m <= count; m++) {
  44. //下方边长 matrix[addRow][j + m]
  45. //右方边长 matrix[i + m][addCol] == '0'
  46. if (matrix[addRow][j + m] == 0
  47. || matrix[i + m][addCol] == 0) {
  48. flag = false;
  49. }
  50. }
  51. if (flag) {
  52. tempMaxSide = Math.max(tempMaxSide, count + 1);
  53. }
  54. else {
  55. //即使右边长、下方边长遇到0. 继续放行验证。
  56. //因为当前边长有可能不是最终正方形的边界
  57. flag = true;
  58. }
  59. }
  60. /**
  61. * 以当前单元格 matrix[i][j] 边长扩展完毕,
  62. * tempMaxSide 以matrix[i][j] 为正方形左上顶点的最大边长为
  63. * maxSide 整个区域之前最大正方形边长为
  64. */
  65. maxSide = Math.max(maxSide, tempMaxSide);
  66. }
  67. }
  68. return maxSide * maxSide;
  69. }
  70. public static void main(String[] args) {
  71. Largest1BorderedSquare_04_leetcode1139暴力解 ss = new Largest1BorderedSquare_04_leetcode1139暴力解();
  72. //int[][] matrix = {{1,1,1}, {1,0,1},{1,1,1}};
  73. int[][] matrix = {{1,1,1}, {1,1,0},{1,1,1},{0,1,1},{1,1,1}};
  74. System.out.println(ss.largest1BorderedSquare(matrix));
  75. }
  76. }

虽然只是暴力解,但是它只花费了5毫秒,胜率86%,已经相当的nice了。

那么,这一题是否有动态规划解法呢,答案是有。

下面来说一下动态规划的思路,以事例1为案例进行分析。

原始数组:

下标 0下标 1下标 2
下标 0111
下标 1101
下标 2111

由下往上,推算每个单元格距离底部的距离。遇到1就累加,遇到0就是0.

每个单元格下方1的数量:

下标 0下标 1下标 2
下标 0313
下标 1202
下标 2111

从右往左,推算出每个单元格右侧1的数量:

下标 0下标 1下标 2
下标 0321
下标 1101
下标 2321

1.  原始数组的0行0列为1,那么右侧有3个1, 下方也有3个1.  正方形的边长必须相等,因此两者取小。 也就是说,如果以原始数组0行0列为正方形的左顶点,那么这个正方形的最大边长肯定是小于等于3的。

2.  然后就是遍历了。最大长度为1,为2,为3,逐步验证。最终确定,最大正方形的的边长,到底是多少。

3. 在步骤1讨论的过程中,我们依旧确定了正方形上方边、左侧边的边长了。那么,下方的边、右侧的边也需要验证的。下方的边,可以通过左下方的顶点右侧1的数量来确定,必须大于步骤2讨论的边长。右侧的边可以通过正方形右上方的顶点来确定,右上方的顶点下方1的数量大于等于步骤2讨论的边长即可。

动态规划代码:

  1. package code04.动态规划专项训练02;
  2. /**
  3. * 力扣 1139 最大的以1为边界的正方形
  4. * https://leetcode.com/problems/largest-1-bordered-square/
  5. */
  6. public class Largest1BorderedSquare_04_leetcode1139动态规划 {
  7. //统计每个单元格,右侧连续有几个1;下方连续有几个1;
  8. public void preHandle(int[][] matrix, int[][] rightArr, int[][] bottomArr)
  9. {
  10. int rowLength = matrix.length;
  11. int colLength = matrix[0].length;
  12. bottomArr[rowLength-1][colLength-1] = matrix[rowLength-1][colLength-1];
  13. rightArr[rowLength-1][colLength-1] = matrix[rowLength-1][colLength-1];
  14. //最后一行.
  15. for (int i = colLength - 2; i >= 0; i--) {
  16. //距离底部有几个1
  17. bottomArr[rowLength - 1][i] = matrix[rowLength - 1][i];
  18. //距离右侧有几个1
  19. rightArr[rowLength - 1][i] = matrix[rowLength - 1][i] == 0 ? 0 : matrix[rowLength - 1][i] + rightArr[rowLength - 1][i + 1];
  20. }
  21. //最后一列
  22. for (int i = matrix.length - 2; i >= 0; i--) {
  23. //距离底部有几个1
  24. bottomArr[i][colLength - 1] = matrix[i][colLength - 1] == 0 ? 0 : matrix[i][colLength - 1] + bottomArr[i + 1][colLength - 1];
  25. //距离右侧有几个1
  26. rightArr[i][colLength - 1] = matrix[i][colLength - 1];
  27. }
  28. for (int i = matrix.length -2; i >= 0; i--) {
  29. for (int j = matrix[0].length - 2; j >= 0; j--) {
  30. if (matrix[i][j] == 1) {
  31. bottomArr[i][j] = matrix[i][j] + bottomArr[i+1][j];
  32. rightArr[i][j] = matrix[i][j] + rightArr[i][j + 1];
  33. }
  34. }
  35. }
  36. }
  37. public int largest1BorderedSquare(int[][] matrix) {
  38. if (matrix == null || matrix.length == 0) {
  39. return 0;
  40. }
  41. int row = matrix.length;
  42. int col = matrix[0].length;
  43. int[][] rightArr = new int[row][col];
  44. int[][] bottomArr = new int[row][col];
  45. //预处理数组
  46. preHandle(matrix, rightArr, bottomArr);
  47. //全局正方形最大边长
  48. int maxSide = 0;
  49. for (int i = 0; i < row ; i++) {
  50. //当前行的每一列都作为正方形的左上角,即起始点
  51. for (int j = 0; j < col; j++) {
  52. //当前格子是否为1,为1才可能成为正方形的起始点
  53. if (matrix[i][j] == 0) {
  54. continue;
  55. }
  56. //整个二维数组中,重要有1出现,那正方形边长至少为1
  57. if (maxSide == 0) {
  58. maxSide = 1;
  59. }
  60. //行边长,列边长,两者取小。因为是正方形
  61. int sideLength = Math.min(rightArr[i][j], bottomArr[i][j]);
  62. boolean flag = true;
  63. for (int m = 1; m < sideLength; m++) {
  64. //正方形的右上顶点往下数,即右边边长
  65. int rightLength = bottomArr[i][j + m];
  66. //正方形的底部边长
  67. int bottomLength = rightArr[i + m][j];
  68. //当前边长
  69. int length = m + 1;
  70. if (rightLength >= length && bottomLength >= length) {
  71. maxSide = Math.max(maxSide, length);
  72. }
  73. }
  74. }
  75. }
  76. return maxSide * maxSide;
  77. }
  78. public static void main(String[] args) {
  79. Largest1BorderedSquare_04_leetcode1139动态规划 ss = new Largest1BorderedSquare_04_leetcode1139动态规划();
  80. //int[][] matrix = {{1,1,1}, {1,0,1},{1,1,1}};
  81. int[][] matrix = {{1,1,1}, {1,1,0},{1,1,1},{0,1,1},{1,1,1}};
  82. System.out.println(ss.largest1BorderedSquare(matrix));
  83. }
  84. }

7毫秒,50%的胜率,也还可以了。

虽然这一题也有动态规划,但是,它是基于暴力解逐步演化过来的。暴力解法时间复杂度为 O(N^4)。我们基于暴力解在讨论边长的过程中还需要一次遍历,通过数组预处理的的方式提前进行了统计,简化了验证新增列的验证。因此,动态规划的时间复杂度为 O(N^3) + O(N^2).  实际就是O(N^3)。

这一道题,动态规划的性能比暴力解的还要低一些,虽然动态规划的时间复杂度为O(N^3),而暴力解的时间复杂度为O(N^4)。

动态规划的时间复杂度为O(N^3),但是,它还有一个O(N^2)预处理数组的过程。因为力扣数据量有限,动态规划时间复杂度虽然降低了一阶,但是直观看起来还是没有暴力解高。

随着数据量的增大,动态规划还是要优于暴力解的。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/喵喵爱编程/article/detail/872061
推荐阅读
相关标签
  

闽ICP备14008679号