首页 > 其他 > 详细

221. Maximal Square

时间:2017-05-14 09:33:52      阅读:293      评论:0      收藏:0      [点我收藏+]

Problem statement:

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.

Solution:

It looks like 85. Maximal Rectangle. But, there is big difference. This is DP problem, however, maximul rectangle needs tricky.

The key point is what we want from DP? The answer is max side length of square.

DP maintains a 2D array, dp[i][j] means the max side of length of square by current position.

Initialization:

dp[0][j] = matrix[0][j] - 0;

dp[i][0] = matrix[i][0] - 0;

DP formula. and update the max side length for each value.

if(matrix[i][j] == 1){
    dp[i][j] = min(dp[i - 1][j - 1], min(dp[i - 1][j], dp[i][j - 1])) + 1;
}

Return max_size * max_size;

The time complexity is O(n * n)

class Solution {
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        if(matrix.empty()){
            return 0;
        }
        int row = matrix.size();
        int col = matrix[0].size();
        vector<vector<int>> square_size(row, vector<int>(col, 0));
        int max_size = 0;
        for(int i = 0; i < row; i++){
            square_size[i][0] = matrix[i][0] - 0;
            max_size = max(max_size, square_size[i][0]);
        }
        for(int j = 0; j < col; j++){
            square_size[0][j] = matrix[0][j] - 0;
            max_size = max(max_size, square_size[0][j]);
        }
        for(int i = 1; i < row; i++){
            for(int j = 1; j < col; j++){
                if(matrix[i][j] == 1){
                    square_size[i][j] = min(square_size[i - 1][j - 1], min(square_size[i - 1][j], square_size[i][j - 1])) + 1;
                }
                max_size = max(max_size, square_size[i][j]);
            }
        }
        return max_size * max_size;
    }
};

 

221. Maximal Square

原文:http://www.cnblogs.com/wdw828/p/6851507.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!