LeetCode每日一题——719.找出第k小的数对距离

LeetCode每日一题——719.找出第k小的数对距离,第1张

文章目录
  • 题目
  • 示例
  • 思路

题目

数对 (a,b) 由整数 a 和 b 组成,其数对距离定义为 a 和 b 的绝对差值。

给你一个整数数组 nums 和一个整数 k ,数对由 nums[i] 和 nums[j] 组成且满足 0 <= i < j < nums.length 。返回 所有数对距离中 第 k 小的数对距离。

示例

示例 1:

输入:nums = [1,3,1], k = 1
输出:0
解释:数对和对应的距离如下:
(1,3) -> 2
(1,1) -> 0
(3,1) -> 2
距离第 1 小的数对是 (1,1) ,距离为 0 。

示例 2:

输入:nums = [1,1,1], k = 2
输出:0

示例 3:

输入:nums = [1,6,1], k = 3
输出:5

提示:

n == nums.length
2 <= n <= 104
0 <= nums[i] <= 106
1 <= k <= n * (n - 1) / 2

思路

排序+二分+枚举

class Solution(object):
    def getCount(self, d, nums, size):
        j,count = 0, 0
        for i in range(1, size):
            while nums[i]-nums[j] > d: j+=1
            count += i-j
        return count
    def smallestDistancePair(self, nums, k):
        nums.sort()
        size = len(nums) 
        l = 0
        r = nums[size-1] - nums[0]
        while l<r:
            mid = (l+r) >> 1
            count = self.getCount(mid, nums, size)
            if count < k: l = mid+1
            else: r = mid
        return r

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/1498175.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-06-25
下一篇 2022-06-25

发表评论

登录后才能评论

评论列表(0条)

保存