21-4-22 矩形区域不超过 K 的最大数值和

363. 矩形区域不超过 K 的最大数值和 难度[困难]

给你一个 m x n 的矩阵 matrix 和一个整数 k ，找出并返回矩阵内部矩形区域的不超过 k 的最大数值和。

题目数据保证总会存在一个数值和不超过 k 的矩形区域。

示例 1：

输入：matrix = [[1,0,1],[0,-2,3]], k = 2
输出：2
解释：蓝色边框圈出来的矩形区域 [[0, 1], [-2, 3]] 的数值和是 2，且 2 是不超过 k 的最大数字（k = 2）。

示例 2：

输入：matrix = [[2,2,-1]], k = 3
输出：3

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 100
• -100 <= matrix[i][j] <= 100
• -10^5 <= k <= 10^5

---

解法一：套用 【二维前缀和】 模板

int[][] sum;
    public int maxSumSubmatrix(int[][] matrix, int k) {
        int n=matrix.length;
        int m= n==0 ? 0 : matrix[0].length;
        sum=new int [n+1][m+1];
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                sum[i][j]= sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+matrix[i-1][j-1];
            }
        }
        int ans =Integer.MIN_VALUE;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                for(int p=i;p<=n;p++){
                    for(int q=j;q<=m;q++){
                        int cur = sum[p][q]-sum[i-1][q]-sum[p][j-1]+sum[i-1][j-1];
                        if(cur<=k){
                            ans=Math.max(cur,ans);
                        }
                    }
                }
            }
        }
        return ans;
    }

解法二：二分， 二维抽象一维

详解

int[][] sum;
public int maxSumSubmatrix1(int[][] matrix, int k) {
        int n=matrix.length;
        int m= n==0 ? 0 : matrix[0].length;
        sum=new int [n+1][m+1];
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                sum[i][j]= sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+matrix[i-1][j-1];
            }
        }
        int ans =Integer.MIN_VALUE;
        //先确定上下边界
        for(int top=1;top<=n;top++){
            for(int bot=top;bot<=n;bot++){
                //用set 存储 已经遍历过的 right ，以便 left在set寻找合适值
                TreeSet<Integer> ts =new TreeSet<>();
                ts.add(0);
                for(int r=1;r<=m;r++){
                    int right =  sum[bot][r]-sum[top-1][r];
                    Integer left = ts.ceiling(right-k) ;
                    if(left!=null){
                        int cur =right-left;
                        ans=Math.max(ans,cur);
                    }
                    //添加遍历过的right
                    ts.add(right);
                }

            }
        }
        return ans;
    }

上面的方法还是可以优化的就是将二分搜收的范围变广，如果宽比长大就扫描宽，反之扫描长

class Solution {
    int[][] sum;
    public int maxSumSubmatrix(int[][] matrix, int k) {
       int n=matrix.length;
        int m= n==0 ? 0 : matrix[0].length;
        sum=new int [n+1][m+1];
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                sum[i][j]= sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+matrix[i-1][j-1];
            }
        }
        int ans =Integer.MIN_VALUE;
        boolean isRight = m>n ;
        //先确定上下边界
       for(int i=1;i<=(isRight?n:m);i++){
            for(int j=i;j<=(isRight?n:m);j++){
                //用set 存储 已经遍历过的 right ，以便 left在set寻找合适值
                TreeSet<Integer> ts =new TreeSet<>();
                ts.add(0);
                for(int fixed=1;fixed<=(isRight?m:n);fixed++){
                    int a = isRight?sum[j][fixed]-sum[i-1][fixed]:sum[fixed][j]-sum[fixed][i-1] ;
                    Integer b = ts.ceiling(a-k) ;
                    if(b!=null){
                        int cur =a-b;
                        ans=Math.max(ans,cur);
                    }
                    //添加遍历过的right
                    ts.add(a);
                }

            }
        }
        return ans;
    }
}

空间优化

在搜索矩阵的时候给sum赋值

public int maxSumSubmatrix(int[][] matrix, int k) {
     int n=matrix.length;
        int m= n==0 ? 0 : matrix[0].length;
        boolean isRight = m>n ;
        int [] sum= isRight ? new int [m+1]:new int[n+1];
        int ans =Integer.MIN_VALUE;
        //先确定上下边界
        for(int i=1;i<=(isRight?n:m);i++){
            Arrays.fill(sum,0);
            for(int j=i;j<=(isRight?n:m);j++){
                //用set 存储 已经遍历过的 right ，以便 left在set寻找合适值
                TreeSet<Integer> ts =new TreeSet<>();
                ts.add(0);
                int a = 0;
                for(int fixed=1;fixed<=(isRight?m:n);fixed++){
                    sum[fixed] += isRight ?matrix[j-1][fixed-1] : matrix[fixed-1][j-1] ;
                    a+=sum[fixed];
                    Integer b = ts.ceiling(a-k) ;
                    if(b!=null){
                        int cur =a-b;
                        ans=Math.max(ans,cur);
                    }
                    //添加遍历过的right
                    ts.add(a);
                }

            }
        }
        return ans;
    }

参考

【宫水三叶】优化枚举的基本思路 & 将二维抽象成一维 & 最大化「二分」效益 & 空间优化

---

此文章创于本人学习时的记录，如有错误或更优解还请指出