python leetcode 363. Max Sum of Rectangle No Larger Than K
自己码了代码TLE 然后研究了别人的代码
成功地将时间复杂度由O(mn)^2降到了nnm*log(m)
这里m是行长列长中大的那个
假如是列较长 那就扫描列 利用bisect进行二分查找,插入(slist)
sums是行叠加后的大小(x到y行 第z列的和) num存储的是x到y行 0到z列的和
核心代码
if num<=k: ans=max(ans,num)
i = bisect.bisect_left(slist,num-k)
if i!=len(slist): ans=max(ans,num-slist[i])
bisect.insort(slist,num)
if num<=k: ans=max(ans,num) 这一句很好理解
i = bisect.bisect_left(slist,num-k) 当num>k时 查找x到y行 某列到z列的和
if i!=len(slist): ans=max(ans,num-slist[i]) 如果i!=slist的长度 即能找到某列
bisect.insort(slist,num) 把当前的和插入到slist中
class Solution:
def maxSumSubmatrix(self, matrix, k):
"""
:type matrix: List[List[int]]
:type k: int
:rtype: int
"""
#加二分查找一次比较sum[:i]和sum[i:]
m=len(matrix)
if m==0:
return 0
n=len(matrix[0])
M = max(m,n)
N = min(m,n)
ans = None
for x in range(N):
sums=[0]*M
for y in range(x,N):
slist,num=[],0
for z in range(M):
#扫描列
sums[z] += matrix[z][y] if m>n else matrix[y][z]
num+=sums[z]
if num<=k: ans=max(ans,num)
i = bisect.bisect_left(slist,num-k)
if i!=len(slist): ans=max(ans,num-slist[i])
bisect.insort(slist,num)
return ans or 0