Leetcode Hot100不熟练题目 85. 最大矩形

1、题目描述

给定一个仅包含 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
1 <= row, cols <= 200
matrix[i][j] 为 ‘0’ 或 ‘1’

2、思路

我们考虑以[i,j]为右下角的最大1矩阵。这些矩阵的最大值就是答案。

对于上例中的矩阵，我们统计所有以[i,j]为结尾的一行中连续1的个数。计算如下

利用这个统计数据统计最大矩阵，就像84题一样，找最大的矩形。

3、代码

class Solution {
public:
    int maximalRectangle(vector<vector<char>>& matrix) {
        int ntx = matrix.size();
        int mtx = matrix[0].size();
        // 存放统计数据，第一行上面和最后一行下面多存一行0，目的跟84题一样，避免计算特殊情况
        vector<vector<int>> t(ntx + 2, vector<int>(mtx, 0));
        for(int i = 1; i <= ntx; i ++)
        {
            for(int j = 0; j < mtx; j ++)
            {
                if(matrix[i - 1][j] == '1') 
                {
                    t[i][j] += (j > 0 ? t[i][j - 1] : 0) + 1;
                }
            }
        }
        // 结果
        int ans = 0;
        //对于每一列进行一次寻找最大矩阵的操作
        for(int j = 0; j < mtx; j ++)
        {
            stack<int> s;
            for(int i = 0; i < ntx + 2; i ++)
            {
                while(!s.empty() && t[i][j] < t[s.top()][j])
                {
                    int top = s.top();
                    s.pop();
                    ans = max(ans, t[top][j] * (i - s.top() - 1));
                }
                s.push(i);
            }
        }
        return ans;
    }
};

参考资料

官方答案