Description
Given a 2D boolean
matrix filled with False
and True
, find the largest rectangle containing all True
and return its area.
Example
Example 1
Input:
[
[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]
]
Output: 6
Example 2
Input: [ [0,0], [0,0] ] Output: 0
思路:使用dp和单调栈进行求解。
public class Solution { /** * @param matrix a boolean 2D matrix * @return an integer */ public int maximalRectangle(boolean[][] matrix) { // Write your code here int m = matrix.length; int n = m == 0 ? 0 : matrix[0].length; int[][] height = new int[m][n + 1]; int maxArea = 0; for (int i = 0; i < m; i++) { for (int j = 0; j < n; j++) { if (! matrix[i][j]) { height[i][j] = 0; } else { height[i][j] = i == 0 ? 1 : height[i - 1][j] + 1; } } } for (int i = 0; i < m; i++) { int area = maxAreaInHist(height[i]); if (area > maxArea) { maxArea = area; } } return maxArea; } private int maxAreaInHist(int[] height) { Stack<Integer> stack = new Stack<Integer>(); int i = 0; int max = 0; while (i < height.length) { if (stack.isEmpty() || height[stack.peek()] <= height[i]) { stack.push(i++); } else { int t = stack.pop(); max = Math.max(max, height[t] * (stack.isEmpty() ? i : i - stack.peek() - 1)); } } return max; } }