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

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 

你可能感兴趣的:(leetcode,python)