- 矩形区域不超过 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;
}
参考
【宫水三叶】优化枚举的基本思路 & 将二维抽象成一维 & 最大化「二分」效益 & 空间优化
此文章创于本人学习时的记录,如有错误或更优解还请指出