给你一个 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
-105 <= k <= 105
进阶:如果行数远大于列数,该如何设计解决方案?
思路:
m行n列的矩阵,子矩阵不超过k的最大数值和。
先求列前缀和,再枚举上下边界,对每行的列前缀和求行前缀和. 原问题就转为一维问题:“给定整数数组和整数k, 计算该数组的最大区间和, 要求区间和不超过k.”
设sum[i] = row[0] + row[1] + … + row[i], i =1,2,3…, 求使得sum[j]- sum[i] <= k , i<=j, 的最大区间和sum[j] - sum[i],也就是求使得sum[i] >= sum[j] - k成立的sum[i]的最小值。
在取件[0, j]中使用二分查找,找到符合条件的i。
code:
# leetcode submit region begin(Prohibit modification and deletion)
from sortedcontainers import SortedList
class Solution:
def maxSumSubmatrix(self, matrix, k: int