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; } };
时间: 2024-12-18 20:36:24