1385. 两个数组间的距离值

合集 - LeetCode 二分法(17)

1.35. 搜索插入位置2024-09-092.704. 二分查找2024-09-093.744. 寻找比目标字母大的最小字母2024-09-094.2529. 正整数和负整数的最大计数2024-09-09

5.1385. 两个数组间的距离值2024-09-09

6.2300. 咒语和药水的成功对数2024-09-097.2389. 和有限的最长子序列2024-09-098.1170. 比较字符串最小字母出现频次2024-09-099.2080. 区间内查询数字的频率2024-09-0910.2563. 统计公平数对的数目2024-09-0911.981. 基于时间的键值存储2024-09-1012.1146. 快照数组2024-09-1013.658. 找到 K 个最接近的元素2024-09-1014.1818. 绝对差值和2024-09-1015.911. 在线选举2024-09-1016.LCP 08. 剧情触发时间2024-09-1017.[模板题] - 300. 最长递增子序列2024-09-13

收起

题目链接	1385. 两个数组间的距离值
思路	二分查找
题解链接	官方题解
关键点	标准库的利用；二分循环不变式（开区间）：nums[left] < target && nums[right] >= target
时间复杂度	$O((n+m)\log m)$
空间复杂度	$O(1)$

代码实现：

class Solution:
    def findTheDistanceValue(self, arr1: List[int], arr2: List[int], d: int) -> int:
        arr2.sort()
        n = len(arr2)
 
        def lower_bound(val):
            left, right = -1, n
            while left+1 < right:
                mid = (left+right)//2
                if arr2[mid] < val:
                    left = mid
                else:
                    right = mid
            return right
 
        cnt = 0
        for x in arr1:
            p = lower_bound(x)
            if (p == n or abs(x - arr2[p]) > d) and (
                p == 0 or abs(x - arr2[p - 1]) > d
            ):
                cnt += 1
 
        return cnt

Python-官方库

class Solution:
    def findTheDistanceValue(self, arr1: List[int], arr2: List[int], d: int) -> int:
        arr2.sort()
        cnt = 0
        for x in arr1:
            p = bisect_left(arr2, x)
            if (p == len(arr2) or abs(x - arr2[p]) > d) and (
                p == 0 or abs(x - arr2[p - 1]) > d
            ):
                cnt += 1
 
        return cnt