719. Find K-th Smallest Pair Distance

解法

主要思想是二分法
先给数组排序,这样答案一定在0和最大差距nums[-1]-nums[0]之间
然后要解决的是二分查找中的判断函数问题,如何判断什么时候往左边找什么时候往右边找,在这个问题中,显然当差距小于等于mid的点对的个数大于等于k个的时候要向左找
如何判断【差距小于等于mid的点对的个数大于等于k个】呢?
采用尺取法,尺取法的使用条件为:(1)单调性,即当某端点固定时,另一端点从左向右移动,或从右向左移动,结果是递增或递减的
(2)查询的结果是大于(等于)某值或者小于(等于)某值
本题固定左端点,不断增加右端点,差距是一直增加的,当增加到比mid大之后不符合条件了,这时符合条件的数目为两坐标相减
然后增加左端点,直到再次符合小于等于mid的条件
不会漏掉点对的原因是,如果[i,j]不符合条件,那么[i,j+1,2,…]都不符合条件,而[i,j-1,2,…]都至少在前面某个符合条件的[i,j-1]里统计过了

class Solution(object):
    def smallestDistancePair(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        nums.sort()
        n = len(nums)
        l = 0
        r = nums[-1]-nums[0]
        def isvalid(sum):
            cnt = 0
            l = 0
            for j, x in enumerate(nums):
                while x - nums[l] > sum:
                    l += 1
                cnt += j-l
                if cnt>=k:
                    return True
            return cnt>=k

        while r-l>0:
            mid = (l+r)>>1
            if isvalid(mid):
                r = mid
            else:
                l = mid+1

        return r
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值