矩形区域不超过k的最大数值和
给定一个非空二维矩阵 matrix 和一个整数 _k_,找到这个矩阵内部不大于 k 的最大矩形和。
示例:
输入: matrix = [[1,0,1],[0,-2,3]], k = 2
输出: 2
解释: 矩形区域 [[0, 1], [-2, 3]] 的数值和是 2,且 2 是不超过 k 的最大数字(k = 2)。
说明:
- 矩阵内的矩形区域面积必须大于 0。
- 如果行数远大于列数,你将如何解答呢?
解题思路:
根据题意,寻找二维数组中所有可以组成的矩形中面积不超过k的最大值,所以必须要求出可能组成的矩形的面积并与k比较求出最终结果。这里为了最终不超时,可以在一下方面进行优化:
1.设置一个数组比较当前列(或者行)下已经扫描过的数的和。
2.设置一个TreeMap,保存当前矩阵长度下,已经扫描过得矩形的面积。同时,TreeMap中有快速查找元素的方法,可以快速查找所找的元素。
1 import java.util.TreeSet;
2
3 public class Solution {
4 public static int maxSumSubmatrix(int[][] matrix, int target) {
5 int row = matrix.length;
6 if(row==0)
7 return 0;
8 int col = matrix[0].length;
9 if(col==0)
10 return 0;
11 int result = Integer.MIN_VALUE;
12 boolean key= col>row?false:true;
13 int m = Math.min(row,col);
14 int n = Math.max(row,col);
15 //一行一行的找
16 for(int i=0;i<m;i++){
17 //找从第i行开始一直到第0行这i+1行的可能组成的矩形长度
18 int[] array = new int[n];//这个矩阵为了保存每一列上第j行到第i行的和
19 for(int j=i;j>=0;j--){
20 TreeSet<Integer> set = new TreeSet<Integer>();//用来保存当前高度下,长度为从0开始到k位置的矩形的结果。理解set的含义是解决此题的关键。
21 set.add(0);
22 int sum=0;
23 for(int k=0;k<n;k++){
24 if(key){
25 array[k]+=matrix[k][j];
26 }
27 else{
28 array[k]+=matrix[j][k];
29 }
30 sum+=array[k];
31 /* 因为要满足 (sum-set中的元素)<=target,
32 * 而且sum-set中的元素的值要尽可能的大,
33 * 所以也就是再求小于等于sum-target中满足条件的元素的最小的一个
34 * 正好TreeSet中提供了这个方法ceil(),可以很方便的找出这个元素
35 */
36 Integer integer = set.ceiling(sum-target);
37 if(integer!=null){
38 result=Math.max(result, sum-integer);
39 }
40 set.add(sum);
41 }
42 }
43 }
44 return result;
45 }
46 }