1385. 两个数组间的距离值

给你两个整数数组 arr1 , arr2 和一个整数 d ,请你返回两个数组之间的 距离值 。

「距离值」 定义为符合此距离要求的元素数目:对于元素 arr1[i] ,不存在任何元素 arr2[j] 满足 |arr1[i]-arr2[j]| <= d 。

 

示例 1:

输入:arr1 = [4,5,8], arr2 = [10,9,1,8], d = 2
输出:2
解释:
对于 arr1[0]=4 我们有:
|4-10|=6 > d=2
|4-9|=5 > d=2
|4-1|=3 > d=2
|4-8|=4 > d=2
所以 arr1[0]=4 符合距离要求

对于 arr1[1]=5 我们有:
|5-10|=5 > d=2
|5-9|=4 > d=2
|5-1|=4 > d=2
|5-8|=3 > d=2
所以 arr1[1]=5 也符合距离要求

对于 arr1[2]=8 我们有:
|8-10|=2 <= d=2
|8-9|=1 <= d=2
|8-1|=7 > d=2
|8-8|=0 <= d=2
存在距离小于等于 2 的情况,不符合距离要求

故而只有 arr1[0]=4 和 arr1[1]=5 两个符合距离要求,距离值为 2
示例 2:

输入:arr1 = [1,4,2,3], arr2 = [-4,-3,6,10,20,30], d = 3
输出:2
示例 3:

输入:arr1 = [2,1,100,3], arr2 = [-5,-2,10,-3,7], d = 6
输出:1
 

提示:

1 <= arr1.length, arr2.length <= 500
-10^3 <= arr1[i], arr2[j] <= 10^3
0 <= d <= 100

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-the-distance-value-between-two-arrays

直接暴

class Solution:
    def findTheDistanceValue(self, arr1: List[int], arr2: List[int], d: int) -> int:
        # arr2.sort()
        res=0
        for i in arr1:
            fl=1
            for j in arr2:
                if abs(i-j)<=d:
                    fl=0
                    break
            if fl==1:
                res+=1
        return res

二分

class Solution:
    def findTheDistanceValue(self, arr1: List[int], arr2: List[int], d: int) -> int:
        arr2.sort()
        res=0
        for i in arr1:
            l,r=0,len(arr2)
            while l<r:
                mid=l+(r-l)//2
                if arr2[mid]<i:
                    l=mid+1
                else:
                    r=mid
            if l==0:
                if arr2[0]-i>d:
                    res+=1
            elif l==len(arr2):
                if i-arr2[-1]>d:
                    res+=1
            else:
                if abs(i-arr2[l])>d and abs(i-arr2[l-1])>d:
                    res+=1
        return res                

骚操作,见

bisect in Python

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

 

posted @ 2020-10-10 20:12  XXXSANS  阅读(210)  评论(0编辑  收藏  举报