给你两个整数数组 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
1 class Solution { 2 public: 3 /* 4 * -d <= arr1[i] - arr2[j] <= d, 假定给定值为arr1[i],求在arr2[j]中是否存在值为arr1[i]满足arr1[i] - d <= arr2[j] <= arr1[i] + d 5 */ 6 bool isExistDistanceVal(vector<int> &arr2, int d, int target) { 7 int left = 0; 8 int right = arr2.size(); 9 while (left <= right) { 10 int mid = left + (right - left) / 2; 11 if (arr2[mid] >= target - d && arr2[mid] <= target + d) { 12 break; 13 } else if (arr2[mid] < target - d) { 14 left = mid + 1; 15 } else if (arr2[mid] > target + d) { 16 right = mid - 1; 17 } 18 } 19 return (left > right); 20 } 21 int findTheDistanceValue(vector<int>& arr1, vector<int>& arr2, int d) { 22 int numOfArr1 = arr1.size(); 23 int numOfArr2 = arr2.size(); 24 sort(arr2.begin(), arr2.end()); 25 int cnt = 0; 26 for (auto &val : arr1) { 27 if (isExistDistanceVal(arr2, d, val)) { 28 cnt++; 29 } 30 } 31 return cnt; 32 } 33 };