【leetcode】Maximal Rectangle (hard)★

Given a 2D binary matrix filled with 0's and 1's, find the largest rectangle containing all ones and return its area.

 

找到01矩形中最大的全1子矩阵。

 

我自己的思路：

      我先用一个跟输入相同大小的矩阵numInCol 存储从当前位置开始向下有多少连续的1。

　　如   

1 0 1
0 1 1
1 1 1

其numInCol 是

1 0 3
0 2 2
1 1 1

然后用一个变量tmpans来记录以某个位置开始，其右下矩形的最大全1矩阵的大小。

方法是如果当前位置是1，则用1 * numInCol[当前位置]     即只有这一列的大小

         如果其后面紧接着的位置也是1，则用 2 * (这两列中连续1高度小的值)  即这两列的矩形大小

         如果其后面紧接着的位置也是1，则用 3 * (这三列中连续1高度小的值)  即这三列的矩形大小

         ........................

         依次类推

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <stack>
using namespace std;

class Solution {
public:
    int maximalRectangle(vector<vector<char> > &matrix) {
        if(matrix.empty())
            return 0;

        int row = matrix.size();
        int col = matrix[0].size();
        int ans = 0;

        vector<vector<int> > numInCol(row, vector<int>(col, 0));
        int tmpans = 0;

        //对每一列
        for(int j = 0; j < col; j++)
        {
            numInCol[row - 1][j] = (matrix[row - 1][j] == '1') ? 1 : 0;
            for(int i = row - 2; i >= 0; i--)
            {
                if(matrix[i][j] == '1')
                {
                    numInCol[i][j] = numInCol[i + 1][j] + 1;
                }
            }
        }

        //对整个矩阵
        for(int i = row - 1; i >= 0; i--)
        {
            tmpans = numInCol[i][col - 1];
            ans = max(ans, tmpans);
            for(int j = col - 2; j >= 0; j--)
            {
                int jj = j;
                int len = 1;
                int minlen = numInCol[i][j];
                while(jj < col && matrix[i][jj] == '1')
                {
                    minlen = min(minlen, numInCol[i][jj]);
                    tmpans = max(tmpans, len * minlen);
                    ans = max(ans, tmpans);
                    jj++;
                    len++;

                }
            }
        }

        return ans;
    }
};

int main()
{
    Solution s;
    vector<vector<char> > matrix(4, vector<char>(5, '0'));
    matrix[0][1] = '1';
    matrix[0][2] = '1';
    matrix[0][3] = '1';
    matrix[1][1] = '1';
    matrix[1][2] = '1';
    matrix[2][1] = '1';
    matrix[2][2] = '1';
    matrix[2][3] = '1';

    int ans = s.maximalRectangle(matrix);

    return 0;
}

 

网上的答案，整体的思路差不多，也是通过一列列的数据来判断，但是并没有像我一样，把所有的值都存下来，而是只存了一行的列信息，每到新的一行再更新，这样空间少了很多。其次，没有像我这样每次都循环后面的列是否为1，而是利用栈，如果高度是递增的就先不计算最大矩形而是把信息压栈，等到遇到高度减少的时候再依次计算这一段的最大矩形。

class Solution {
public:
    int maximalRectangle(vector<vector<char>> &matrix) {
        if (matrix.empty()) return 0;
        int rows = matrix.size();
        int cols = matrix[0].size();
        int maxArea = 0;
        vector<int> heights(cols + 1, 0);
        vector<int> stack;
        for (int i = 0; i < rows; i++) {
            stack.clear();
            for (int j = 0; j < cols + 1; j++) {
                if (j < cols) {
                    if (matrix[i][j] == '1') {
                        heights[j]++;
                    } else {
                        heights[j] = 0;
                    }
                }
                if (stack.empty() || heights[j] >= heights[stack.back()]) {
                    stack.push_back(j);
                    continue;
                }
                while (!stack.empty() && heights[j] < heights[stack.back()]) {
                    int top = stack.back();
                    stack.pop_back();
                    int begin = stack.empty() ? 0 : stack.back() + 1;
                    int area = heights[top] * (j - begin);
                    if (area > maxArea) maxArea = area;
                }
                stack.push_back(j);
            }
        }
        return maxArea;
    }
};