【困难】求最大子矩阵的大小－Java

分享一个大牛的人工智能教程。零基础！通俗易懂！风趣幽默！希望你也加入到人工智能的队伍中来！请轻击人工智能教程

package live.every.day.ProgrammingDesign.CodingInterviewGuide.StackAndQueue;

import java.util.Stack;

/**
 * 求最大子矩阵的大小
 *
 * 【题目】
 * 给定一个整型矩阵map，其中的值只有0和1两种，求其中全是1的所有矩形区域中，最大的矩形区域为1的数量。
 *
 * 【难度】
 * 困难
 *
 * 【解答】
 * 如果矩阵的大小为O(NxM)，本题可以做到时间复杂度为O(NxM)。解法的具体过程为：
 * １、矩阵的行数为N，以每一行做切割，统计以当前行作为底的情况下，每个位置往上的1的数量。使用高度数组height来表示。
 * ２、对于每一次切割，都利用更新后的height数组来求出以每一行为底的情况下，最大的矩形是什么。那么这么多次切割中，最大的
 * 那个矩形就是我们要的。
 *
 * 整个过程就是如下代码中的getMaxRectangleSize方法。步骤2中的实现是如下代码中的getMaxRectangleFromBottom方法。
 *
 * 下面重点介绍一下步骤2如何快速实现，这也是这道题最重要的部分，如果height数组的长度为M，那么求解步骤2的过程可以做到时
 * 间复杂度为O(M)。
 *
 * 对于height数组，可以理解为一个直方图。
 * 也就是说，步骤2的实质是在一个大的直方图中求最大矩形的面积。如果我们能够求出以每一根柱子扩展出去的最大矩形，那么其中最
 * 大的矩形就是我们想要找的。
 * 考查每一根柱子最大能扩多大，这个行为的实质就是找到柱子左边刚比它小的柱子位置在哪里，以及右边刚比它小的位置在哪里。这个
 * 过程怎么计算最快呢？用栈。
 *
 * @author Created by LiveEveryDay
 */

public class MaxSubMatrixSize {

    public static int getMaxRectangleSize(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[] items : map) {
            for (int j = 0; j < map[0].length; j++) {
                height[j] = items[j] == 0 ? 0 : height[j] + 1;
            }
            maxArea = Math.max(getMaxRectangleFromBottom(height), maxArea);
        }
        return maxArea;
    }

    private static int getMaxRectangleFromBottom(int[] height) {
        if (height == null || height.length == 0) {
            return 0;
        }
        int maxArea = 0;
        Stack<Integer> stack = new Stack<>();
        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 currentArea = (i - k - 1) * height[j];
                maxArea = Math.max(maxArea, currentArea);
            }
            stack.push(i);
        }
        while (!stack.isEmpty()) {
            int j = stack.pop();
            int k = stack.isEmpty() ? -1 : stack.peek();
            int currentArea = (height.length - k - 1) * height[j];
            maxArea = Math.max(maxArea, currentArea);
        }
        return maxArea;
    }

    public static void main(String[] args) {
        int[][] map = {{1, 0, 1, 1}, {1, 1, 1, 1}, {1, 1, 1, 0}};
        System.out.printf("The max sub matrix size is: %d", getMaxRectangleSize(map));
    }

}

// ------ Output ------
/*
The max sub matrix size is: 6
*/