LeetCode第221题--最大正方形

最新推荐文章于 2021-04-15 17:31:20 发布

king3wade

最新推荐文章于 2021-04-15 17:31:20 发布

阅读量229

点赞数

分类专栏： LeetCode 算法编程学习文章标签： LeetCode 最大正方形

本文链接：https://blog.csdn.net/king3wade/article/details/92021822

版权

LeetCode 算法编程学习专栏收录该内容

95 篇文章 0 订阅

订阅专栏

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

示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

输出: 4

class Solution {
    int getMaxSquare(vector<int> &nums)
    {
        if(nums.size() == 0) return 0;
        stack<int> sta;
        int maxArea = 0;
        vector<int> vecTmp(nums.begin(), nums.end());
        vecTmp.push_back(-1);
        for(int i = 0; i < vecTmp.size(); i++)
        {
            while(!sta.empty() && vecTmp[i] < vecTmp[sta.top()])
            {
                int height = vecTmp[sta.top()];
                sta.pop();
                int width = (sta.empty() ? i : i-sta.top()-1);
                int squareLen = min(height, width);		//确保按正方形计算
                maxArea = max(maxArea, squareLen*squareLen);
            }
            sta.push(i);
        }
        return maxArea;
    }
    
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        if(matrix.size() == 0 || matrix[0].size() == 0) return 0;
        int rowSize = matrix.size();
        int colSize = matrix[0].size();
        vector<int> vecTmp(colSize, 0);
        int maxArea = 0;
        for(int i = 0; i < rowSize; i++)
        {
            for(int j = 0; j < colSize; j++)
            {
                if(matrix[i][j] == '0')
                {
                    vecTmp[j] = 0;
                }
                else
                {
                    vecTmp[j]++;
                }
            }
            maxArea = max(maxArea, getMaxSquare(vecTmp));
        }
        return maxArea;
    }
};

这个题是第84题柱状图的最大矩形和85题最大矩形的进阶版，采用单调栈求出最大正方形，注意限制条件squareLen = min(height, width)，然后从上向下遍历，遇到’0’清空，否则累加，从而求出最大正方形。