动态规划:01矩阵中全为1的最大正方形边长

问题描述

在一个二维0 1矩阵中找到全为1的最大正方形,返回其边长。

输入:

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

输出:

2

思路:

动态规划,构造一个等大小的dp[n][m]矩阵,dp[i][j]表示以matrix[i][j]为右下角的全1正方形的边长。初始化dp[i][0]dp[0][j]分别为matrix[i][0]matrix[0][j],因为以边缘这一行一列中的点为右下角的全1正方形边长只有01两种情况,分别由他们本身是0还是1决定。初始化之后开始遍历这个dp矩阵,如果matrix[i][j]0,则dp[i][j]1,如果matrix[i][j]1,则dp[i][j]=min{dp[i-1][j],dp[i][j-1],dp[i-1][j-1]}+1,这一步可以这样理解:

矩阵示意图

只有满足matrix[i][j]左方上方和左上方三个地方的正方形边长为2的时候,它的边长才能为3,任意一个为1,则它的边长为2,所以取三者最小值+1就是以该点为右下角的全1正方形边长。最后只用遍历dp矩阵找出最大值即可。

代码:

public class Solution {
    public int FindAllOneSquare(int [][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        int dp[][] = new int[m][n];
        for (int i = 0; i < m; i++){
            dp[i][0] = matrix[i][0];
        }
        for (int j = 0; j < n;j++){
            dp[0][j] = matrix[0][j];
        }
        for (int i=1; i < m; i++){
            for (int j = 1; j < n; j++){
                if (matrix[i][j] == 0)
                    dp[i][j] = 0;
                else {
                    dp[i][j] = Math.min(dp[i-1][j-1],
                            Math.min(dp[i][j-1], dp[i-1][j])) + 1;
                }
            }
        }
        int max = 0;
        for (int i = 0; i < m; i++){
            for (int j = 0; j < n; j++){
                if (dp[i][j] > max)
                    max = dp[i][j];
            }
        }
        return max;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 题目链接tag: Medium; Dynamic Programming; question:  Given a ...
    xingzai阅读 3,345评论 0 1
  • 回溯算法 回溯法:也称为试探法,它并不考虑问题规模的大小,而是从问题的最明显的最小规模开始逐步求解出可能的答案,并...
    fredal阅读 14,672评论 0 89
  • 江湖中的镖局 早已无从寻找 只余下一湾碧水绕城南流 去往潇湘的渡船 何时启程 载着一江倒映的灯光 一行柳岸的清风 ...
    最爱微风草动摇阅读 1,872评论 0 1
  • 智齿,是指牙槽骨最里面的上下左右各一颗的第三磨牙,也就是从外往里数第三颗大牙。一般在青春期以后开始长,被看做是“智...
    一名小牙医阅读 5,377评论 0 0
  • 晚上你要知道明天上班所以无论你做什么你应该清楚什么状态才是你该有的!所以工作加一个要素叫准备。 自信,认真,勤劳,...
    H希O望PE阅读 984评论 0 0