最大矩形-LintCode

来源:互联网 发布:淘宝实名认证手机 编辑:程序博客网 时间:2024/05/22 09:49

给你一个二维矩阵,权值为False和True,找到一个最大的矩形,使得里面的值全部为True,输出它的面积

样例:
给你一个矩阵如下

[  [1, 1, 0, 0, 1],  [0, 1, 0, 0, 1],  [0, 0, 1, 1, 1],  [0, 0, 1, 1, 1],  [0, 0, 0, 0, 1]]

输出6

思路:
参考题目“直方图最大矩形覆盖”,将矩阵的每一行当做是直方图来求解。

#ifndef C510_H#define C510_H#include<iostream>#include<vector>#include<stack>using namespace std;class Solution {public:    /*    * @param matrix: a boolean 2D matrix    * @return: an integer    */    int maximalRectangle(vector<vector<bool>> &matrix) {        // write your code here        if (matrix.empty())            return 0;        int row = matrix.size();        int col = matrix[0].size();        vector<vector<int>> v(row, vector<int>(col, 0));        for (int j = 0; j < col; ++j)        {            for (int i = 0; i < row; ++i)            {                if (matrix[i][j])                {                    v[i][j] = 1;                    if (i>0)                        v[i][j] += v[i - 1][j];                }                       }        }        int area = 0;        for (auto &c : v)        {            area = maxVal(largestRectangleArea(c), area);        }        return area;    }    int largestRectangleArea(vector<int> &height) {        int res = 0;        stack<int> s;        height.push_back(0);        for (int i = 0; i < height.size(); ++i) {            if (s.empty() || height[s.top()] < height[i]) s.push(i);            else {                int cur = s.top();                s.pop();                res = maxVal(res, height[cur] * (s.empty() ? i : (i - s.top() - 1)));                --i;            }        }        return res;    }    int maxVal(int a, int b)    {        return a > b ? a : b;    }};#endif
原创粉丝点击