左神算法:求最大子矩阵的大小(Java版)

本题来自左神《程序员面试代码指南》“求最大子矩阵的大小”题目。

题目

给定一个整型矩阵 map,其中的值只有0和1两种,求其中全是1的所有矩形区域中,最大的矩形区域为1的数量。
 
例如:

1 0 1 1
1 1 1 1
1 1 1 0
 
其中,最大的矩形区域有 6 个 1,所以返回 6。

题解

本题利用 单调栈 的思想。

关于单调栈,前面我们讲过。可以参考这篇博客 左神算法:单调栈结构(Java版)
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

过程草稿图:
在这里插入图片描述
在这里插入图片描述

代码

import java.util.Stack;

public class Main {

    public static void main(String[] args) {
        int[][] map = {{1, 0, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 0},};
        System.out.println(maxRecSize(map));
    }

    public static int maxRecSize(int[][] map) {
        if (map == null || map.length == 0 || map[0].length == 0) {
            return 0;
        }
        int maxArea = 0;
        int[] height = new int[map[0].length];
        for (int i = 0; i < map.length; i++) { // 以第i行做切割
            for (int j = 0; j < map[0].length; j++) { // 第j列位置往上的1的数量,即高度
                height[j] = map[i][j] == 0 ? 0 : height[j] + 1;
            }
            maxArea = Math.max(maxRecFromBottom(height), maxArea);
        }
        return maxArea;
    }

    /**
     * 计算以当前行做切割(以当前行为底)的情况下,最大的矩形是什么
     * @param height 更新后的 height 数组
     * @return
     */
    public static int maxRecFromBottom(int[] height) {
        if (height == null || height.length == 0) {
            return 0;
        }
        int maxArea = 0;
        Stack<Integer> stack = new Stack<Integer>(); // 单调栈
        for (int i = 0; i < height.length; i++) {
            while (!stack.isEmpty() && height[i] <= height[stack.peek()]) {
                int j = stack.pop();
                int k = stack.isEmpty() ? -1 : stack.peek();
                int curArea = (i - k - 1) * height[j]; // =((i-1)-(k+1)+1)*height[j],即,j 向右至少能扩展到 i-1,向左能精确扩展到 k+1,计算 (右边界-左边界+1)*高
                maxArea = Math.max(maxArea, curArea);
            }
            stack.push(i);
        }
        // 遍历结束,stack中仍有剩余的元素未经过扩展。需要将它们依次弹出,并计算扩展。
        while (!stack.isEmpty()) {
            int j = stack.pop();
            int k = stack.isEmpty() ? -1 : stack.peek();
            int curArea = (height.length - k - 1) * height[j]; // 认为 i==leight.length
            maxArea = Math.max(maxArea, curArea);
        }
        return maxArea;
    }
}

输出:6

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值