前端算法其一四六期-1385. 两个数组间的距离值

114 阅读1分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第2天,点击查看活动详情

给你两个整数数组 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<=5001 <= arr1.length, arr2.length <= 500
  • 103<=arr1[i],arr2[j]<=103-10^3 <= arr1[i], arr2[j] <= 10^3
  • 0<=d<=1000 <= d <= 100

模拟

我们可以对 arr2 进行排序,然后对于 arr1 中的每个元素 xi,在 arr2 中二分寻找上述的两个 yj,如果这两个元素满足性质,则所有元素都满足性质,将答案增加 1。

代码如下。

var findTheDistanceValue = function(arr1, arr2, d, n = 0) {
    arr2.sort((a, b) => a - b)
    var binary = (v) => {
        var l = 0, r = arr2.length - 1
        while(l <= r) {
            var m = l + (r - l >> 1), t = arr2[m]
            if (t === v) return m
            else if (t > v) r = m - 1
            else l = m + 1
        }
        return l
    }
    for (var v of arr1) {
        var i = binary(v), f = true
        if (arr2[i] - v <= d || i !== 0 && v - arr2[i - 1] <= d) f = false
        if (f) n++
    }
    return n
};

图片.png