7月14日 简单的排序二分搜索题,活用 bisect和 sortedcontainers的题目

今天沉迷于看小说所以就做了一道题，实在惭愧

1818. 绝对差值和

首先是最基础的写法，在这个写法中用到了二分搜索，我使用bisect取代二分搜索

class Solution:
    def minAbsoluteSumDiff(self, nums1: List[int], nums2: List[int]) -> int:
        #排序加二分搜索题目
        import bisect
        sort_n1 = sorted(nums1)
        n,res,max_= len(nums1),0,0
        for i in range(n):
            absn = abs(nums1[i]-nums2[i])
            indx = bisect.bisect_left(sort_n1,nums2[i])
            if indx == n:
                min_res = nums2[i]- sort_n1[indx-1]
            elif indx == 0:
                min_res = sort_n1[indx]- nums2[i]
            else:
                min_res = min(sort_n1[indx]- nums2[i],nums2[i]- sort_n1[indx-1])
            res+=absn
            max_ = max(max_,absn-min_res)
            #print(min_res)
        return (res-max_)%(10 ** 9 + 7)

bisect_left 当所找的数字在list中的时候，会返回那个数字的索引，这里因为边界条件加了些判断

为了简介写法，在开始的sorted加入极值，避免边界条件的出现

class Solution:
    def minAbsoluteSumDiff(self, nums1: List[int], nums2: List[int]) -> int:
        #排序加二分搜索题目
        import bisect
        sort_n1 = sorted(nums1+[-inf,inf])
        n,res,max_= len(nums1),0,0
        for i in range(n):
            absn = abs(nums1[i]-nums2[i])
            indx = bisect.bisect_left(sort_n1,nums2[i])
            min_res = min(sort_n1[indx]- nums2[i],nums2[i]- sort_n1[indx-1])
            res+=absn
            max_ = max(max_,absn-min_res)
            #print(min_res)
        return (res-max_)%(10 ** 9 + 7)

之后再使用，sorted list，这个库中有bisect的语法

from sortedcontainers import SortedList
class Solution:
    def minAbsoluteSumDiff(self, nums1: List[int], nums2: List[int]) -> int:
        #排序加二分搜索题目
        sort_n1 = SortedList(nums1+[-inf,inf])
        n,res,max_= len(nums1),0,0
        for i in range(n):
            absn = abs(nums1[i]-nums2[i])
            indx = sort_n1.bisect_left(nums2[i])
            min_res = min(sort_n1[indx]- nums2[i],nums2[i]- sort_n1[indx-1])
            res+=absn
            max_ = max(max_,absn-min_res)
            #print(min_res)
        return (res-max_)%(10 ** 9 + 7)

这个写法速度很慢，可能是调用了多的库，他与上面一个速度为342，一个为1532，相差五倍。

但是当这两种写法同时加上一个判断，速度都会降到152ms

class Solution:
    def minAbsoluteSumDiff(self, nums1: List[int], nums2: List[int]) -> int:
        #排序加二分搜索题目
        import bisect
        sort_n1 = sorted(nums1+[-inf,inf])
        n,res,max_= len(nums1),0,0
        for i in range(n):
            absn = abs(nums1[i]-nums2[i])
            res+=absn
            if absn> max_:
                indx = bisect.bisect_left(sort_n1,nums2[i])
                min_res = min(sort_n1[indx]- nums2[i],nums2[i]- sort_n1[indx-1])
                max_ = max(max_,absn-min_res)
            #print(min_res)
        return (res-max_)%(10 ** 9 + 7)

纠结死了，不知道为啥，在leetcode上问了，希望有结果吧。