LeetCode 85. 最大矩形【单调栈】

给定一个仅包含 0 和 1 、大小为 rows x cols 的二维二进制矩阵，找出只包含 1 的最大矩形，并返回其面积。

示例 1：

输入：matrix = [[“1”,“0”,“1”,“0”,“0”],[“1”,“0”,“1”,“1”,“1”],[“1”,“1”,“1”,“1”,“1”],[“1”,“0”,“0”,“1”,“0”]]
输出：6
解释：最大矩形如上图所示。

示例 2：

输入：matrix = []
输出：0

示例 3：

输入：matrix = [[“0”]]
输出：0

示例 4：

输入：matrix = [[“1”]]
输出：1

示例 5：

输入：matrix = [[“0”,“0”]]
输出：0

提示：

rows == matrix.length
cols == matrix[0].length
0 <= row, cols <= 200
matrix[i][j] 为 '0' 或 '1'

把每一行及其当前行以上部分看成一个柱状图。枚举每一行，求当前行及其以上部分的最大矩形。最大值就是答案。

class Solution {
public:
    static const int maxn = 210;
    int hei[maxn], ans = 0;
    int st[maxn];
    int lef[maxn];
    int rig[maxn];
    int n;
    
    int getAns() {//求每一行及其以上部分的最大矩形
        int top = 0;
        st[top] = -1;
        for(int i = 0; i < n; i++) {
            while(top && hei[st[top]] >= hei[i])    top--;
            lef[i] = st[top];
            st[++top] = i;
        }
        top = 0;
        st[top] = n;
        for(int i = n - 1; i >= 0; i--) {
            while(top && hei[st[top]] >= hei[i]) top--;
            rig[i] = st[top];
            st[++top] = i;
        }
        int mx = 0;
        int ans = 0;
        for(int i = 0; i < n; i++) {
            if(hei[i] == 0) continue;
            int l = i - lef[i] - 1;
            int r = rig[i] - i - 1;
            ans = max(ans, hei[i] * (l + r + 1));
        }
        return ans;
    }
    
    int maximalRectangle(vector<vector<char>>& matrix) {
        if(matrix.size() == 0)  return 0;
        n = matrix[0].size();
        for(int i = 0; i < matrix.size(); i++) {//枚举每一行
            for(int j = 0; j < matrix[i].size(); j++) {
                if(matrix[i][j] == '1') {
                    if(i == 0)  hei[j] = 1;
                    else    hei[j]++;//求出柱状图
                }
                else    hei[j] = 0;
            }
            ans = max(ans, getAns());
        }
        return ans;
    }
};