不积跬步,无以至千里;不积小流,无以成江海。——荀子

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

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.

题目大意:给定一个二维矩阵,值只有0和1。找出最大的只包含1的正方形,然后返回其面积。

思路:使用动态规划。设dp[i][j]为到达点[i][j]时,最大正方形的边长,该正方形需要包含点[i][j]。

位于第0行的点,可以知道当matrix[0][i]='0'时,dp[0][i]=0;当matrix[0][i]='1'时,dp[0][i]=1;位于第0列的同理。

当i>0 and j>0时,当matrix[i][j]='0'时,dp[0][i]=0;当matrix[i][j]='1'时,dp[i][j]=min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1])

class Solution {
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        int n = matrix.size();
        if(n==0) return 0;
        int m = matrix[0].size();
        
        vector<vector<int>> dp(n, vector<int>(m,0));
        int maxSize=0;//最大正方形的边长
        for(int i=0;i<n;i++)//第0行 初始化
        {
            dp[i][0]=matrix[i][0]-'0';
            maxSize = max(maxSize, dp[i][0]);
        }
            
        for(int i=1;i<m;i++)//第0列 初始化
        {
            dp[0][i] = matrix[0][i]-'0';
            maxSize = max(maxSize, dp[0][i]);
        }
            
        
        for(int i=1;i<n;i++)
        {
            for(int j=1;j<m;j++)
            {
                if(matrix[i][j]=='1')
                {
                    dp[i][j] = min(dp[i-1][j], min(dp[i][j-1], dp[i-1][j-1]))+1;
                    maxSize = max(maxSize, dp[i][j]);
                }
            }
        }
        return maxSize*maxSize;
    }
};

 参考:https://discuss.leetcode.com/topic/15328/easy-dp-solution-in-c-with-detailed-explanations-8ms-o-n-2-time-and-o-n-space

posted on 2017-10-01 20:42  hejunlin  阅读(266)  评论(0编辑  收藏  举报