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
二、方法一
只要找到大于等于xi 的第一个 yj 和小于 x 的第一个 y_j ,看看它们满不满足这个性质就可以了
class Solution {
public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
Arrays.sort(arr2);
int res = 0;
for (int i : arr1) {
int p = binSearch(arr2, i);
boolean ok = true;
if (p < arr2.length) {
ok &= arr2[p] - i > d;
}
if (p - 1 >= 0 && p - 1 <= arr2.length) {
ok &= i - arr2[p - 1] > d;
}
res += ok ? 1 : 0;
}
return res;
}
public int binSearch(int[] arr, int target) {
int l = 0;
int r = arr.length - 1;
if (arr[r] < target) {
return r + 1;
}
while (l < r) {
int mid = l + ((r - l) >> 1);
if (arr[mid] > target) {
r = mid;
} else {
l = mid + 1;
}
}
return l;
}
}
复杂度分析
假设 arr1 中元素个数为 n,arr2 中元素个数为 m。
-
时间复杂度:给 arr2 排序的时间代价是 O(mlogm),对于 arr1 中的每个元素都在 arr2 中二分的时间代价是O(nlogm),故渐进时间复杂度为 O((n+m)logm)。
-
空间复杂度:这里没有使用任何的辅助空间,故渐进空间复杂度为 O(1)。
三、方法二
模拟
class Solution {
public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
int res = 0;
for (int i : arr1) {
boolean flag = true;
for (int j : arr2) {
flag &= Math.abs(i - j) > d;
}
res += flag ? 1 : 0;
}
return res;
}
}
复杂度分析
-
时间复杂度:从代码可以看出这里的渐进时间复杂度是 O(n×m)。
-
空间复杂度:这里没有使用任何的辅助空间,故渐进空间复杂度为 O(1)。