Leetcode - Max Sum of Rectangle No Larger Than K

My code:

import java.util.TreeSet;

public class Solution {
    public int maxSumSubmatrix(int[][] matrix, int k) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            return 0;
        }
        
        int max = Integer.MIN_VALUE;
        int m = matrix.length;
        int n = matrix[0].length;
        for (int i = 0; i < n; i++) {
            int[] sum = new int[m];
            for (int j = i; j < n; j++) {
                for (int p = 0; p < m; p++) {
                    sum[p] += matrix[p][j];
                }
                
                int cum = 0;
                TreeSet set = new TreeSet();
                set.add(0);
                for (int p = 0; p < m; p++) {
                    cum += sum[p];
                    Integer value = set.ceiling(cum - k);
                    if (value != null) {
                        if (max < cum - value) {
                            max = cum - value;
                        }
                    }
                    set.add(cum);
                }
            }
        }
            
        return max;
    }
    
    public static void main(String[] args) {
        Solution test = new Solution();
        int[][] matrix = new int[][]{{2,2,-1}};
        int ret = test.maxSumSubmatrix(matrix, 0);
        System.out.println(ret);
    }
}

reference:
https://discuss.leetcode.com/topic/48875/accepted-c-codes-with-explanation-and-references

自己没做出来,一开始就放弃了。的确很难。
这道题目可以拆成至少两道题,最多三道题。

1 . 给你一个 matrix, 找出和最大的那个矩形,返回最大和,以及这个矩形的边界。
https://www.youtube.com/watch?v=yCQN096CwWM
这个阿三的视频讲的实在是太棒了。
time complexity: O(col * col * row)
space complexity: O(col)
思想就是DP,以列为单位,一列一列的累加,同时,对于累加起来的这一列。我们从上到下扫描他,找出其中连续的一段子数列,他的和最大。这就是大矩形下的一个子矩形, 他的和最大。
再和当前最大的和比较,如果更大,那么就更新当前最大和,同时更新当前最大矩阵的边界。

2 . 给你一个一维 array, 找出其中连续的一段,其和最大,但是不大于 k
https://www.quora.com/Given-an-array-of-integers-A-and-an-integer-k-find-a-subarray-that-contains-the-largest-sum-subject-to-a-constraint-that-the-sum-is-less-than-k

time complexity: O(n log n)
space: O(n)

我需要一个 TreeSet 的配合。
我有个变量 cum 用来累加每一个元素的和。然后去询问 Treeset,
.ceiling(cum - k), 是否存在这一个累加和, cum - 它之后,的值,小于等于 k
如果存在,拿出来,算出和 cum的差值,这就是其中一段子数列的和,并且这个和,小于等于 k
然后不断更新这个最大值。

3 . 这道题目,如果扫描列,
time complexity: O(col * col * row * log(row))
space complexity: O(col)
当 col 远大于 row 时,这么做就不合理了。
有两种方法来解决。
第一,直接按行来扫描。
第二,顺时针旋转整个矩阵 90度,然后拿原算法进行计算。

这道题目差不多就介绍到这里了。挺有意思的。

By the way, KMP 算法还记得是什么吗?

Anyway, Good luck, Richardo! -- 09/21/2016

你可能感兴趣的:(Leetcode - Max Sum of Rectangle No Larger Than K)