python leetcode 363. Max Sum of Rectangle No Larger Than K

221 篇文章 2 订阅

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 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值