LeetCode - 1727 交换列得到面积最大子矩阵【排序】

1727. 重新排列后的最大子矩阵

示例2：

输入：matrix = [[1,0,1,0,1]]
输出：3
解释：你可以按照上图方式重新排列矩阵的每一列。
最大的全 1 子矩阵是上图中加粗的部分，面积为 3 。

示例 3：

输入：matrix = [[1,1,0],[1,0,1]]
输出：2
解释：由于你只能整列整列重新排布，所以没有比面积为 2 更大的全 1 子矩形。

思路

  题意：通过交换列，求面积最大的矩形。

  方法：依次计算每个元素在列方向连续1的个数。

如将

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

[1, 1, 1]    转换为     [1, 1, 2]

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

随后，逐行进行排序（便于统计最大矩形）。因为已经转换为连续1的个数，行之间的排序相当于列之间的交换。

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

[1, 1, 2]    转换为     [1, 1, 2]

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

逐行从小到大判断元素大小，若元素num[i]非0，则计算以num[i]为宽、以(num.length - i)为长的矩形大小，保留最大的为最终结果。

代码

class Solution {
    public int largestSubmatrix(int[][] matrix) {
        int result = 0;
        for(int i = 1; i < matrix.length; i++)
            for(int j = 0; j < matrix[0].length; j++)
                if(matrix[i][j] == 1)
                    matrix[i][j] += matrix[i-1][j];
        for (int[] num: matrix) {
            Arrays.sort(num);
            for (int i = 0; i < matrix[0].length; i++) 
                if (num[i] != 0) 
                    result = Math.max(result, num[i] * (num.length - i));
        }
        return result;
    }
}

运行结果