题目
给你两个整数数组 arr1 , arr2 和一个整数d, 请你返回两个数组之间的 距离值。
[距离值」定义为符合此距离要求的元素数目:
对于元素 arr1[i] ,不存在任何元素 arr2[j], 满足 |arr1[i]-arr2[j]| <= d 。
示例:
输入: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
解法一:暴力
对于我这种写法,其实不用排序。(我之前还排了序)
直接两重循环,看是否存在两数之差的绝对值<=d的情况,有则break。
public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
int count = 0;
for (int i = 0; i < arr1.length; i++) {
boolean flag = true;
for (int j = 0; j < arr2.length; j++) {
if (Math.abs(arr1[i] - arr2[j]) <= d) {
flag = false;
break;
}
}
if (flag) {
count++;
}
}
return count;
}
暴力改进版
按照题意模拟:对于arr1 数组中的每一个元素 xi,枚举数组arr2 中的每一个元素 yj,检查是否对于每一个yj 都有|xi-yj|>d,如果是,就将答案增加1。
public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
int cnt = 0;
for (int x : arr1) {
boolean flag = true;
for (int y : arr2) {
flag &= Math.abs(y - x) > d;
}
cnt += flag ? 1 : 0;
}
return cnt;
}
解法二:二分查找
提示:Sort ‘arr2’ and use binary search to get the closest element for each ‘arr1[i]’, it gives a time complexity of O(nlogn).
我一开始就是按照提示来写的,但是我发现我对二分的理解不够,所以我挣扎了一会,我就放弃了。
现在重新来!(失败不可怕,永远努力即可)
分析为什么可以用二分查找?
距离值是:对于数组1中的某个元素,如果遍历数组2,都找不到一个数与之相减的绝对值<=d,则距离值+1.
我们可以固定数组1中的数,然后去数组2中里找值 最接近该数的元素的下标。
不过先二分,先排序。
// 二分查找
int total;
public int findTheDistanceValue2(int[] arr1, int[] arr2, int d) {
Arrays.sort(arr2);
for (int i = 0; i < arr1.length; i++) {
binarySearch(arr2, arr1[i], d);
}
return total;
}
public void binarySearch(int[] arr2, int target, int d) {
int left = 0, right = arr2.length - 1;
while (left + 1 < right) {
int mid = left + (right - left) / 2;
if (arr2[mid] >= target) {
right = mid;
} else if (arr2[mid] < target) {
left = mid;
}
}
int a = Math.abs(target - arr2[right]);
int b = Math.abs(target - arr2[left]);
if ((a <= b && a > d) || (a > b && b > d)) {
total++;
}
}
反思
学会二分来做题啦
官方的二分写法(我在二分的时候直接判断了)
public int binarySearch(int[] arr, int target) {
int low = 0, high = arr.length - 1;
//边界判断的好
if (arr[high] < target) {
return high + 1;
}
while (low < high) {
int mid = (high - low) / 2 + low;
if (arr[mid] < target) {
low = mid + 1;
} else {
high = mid;
}
}
return low;
}
end.