1385. 两个数组间的距离值

1385. 两个数组间的距离值https://leetcode.cn/problems/find-the-distance-value-between-two-arrays/

 

难度简单49

给你两个整数数组 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

通过次数22,398提交次数33,568

class Solution {
    public int findTheDistanceValue(int[] arr1, int[] arr2, int d) {
        // 思路：
        // 先对arr2排序
        // 查找每个arr1[i]在arr[2]中的位置 mid (二分查找)
        // 若 arr1[i]-d > arr2[mid-1] 且 arr1[i]+d <arr2[mid+1] 则ans++;

        //实现
        int ans =0 ;
        Arrays.sort(arr2);
        for(int i=0;i<arr2.length;i++) System.out.print(arr2[i]+"  ");
        System.out.println();

        for(int i=0;i<arr1.length;i++)
        {
            int mid = erfen_find(arr2,arr1[i]);
            System.out.println(arr1[i]+"  "+mid);
            if(mid>0 && arr1[i]>arr2[mid-1]+d && mid<arr2.length && arr1[i]+d<arr2[mid])
            {
                System.out.println(arr1[i]);
                ans++;
            }
            if(mid>=arr2.length && arr1[i]>arr2[mid-1]+d) ans++;
            if(mid<=0 && arr1[i]+d<arr2[0]) ans++;

        }
        return ans;
    }
    
    int erfen_find(int[] arr2,int n)
    {
        int start = 0;
        int end = arr2.length-1;
        int mid;
        while(start<=end)
        {
            mid = (end-start)/2+start;
            // System.out.print(mid+" ");
            if(arr2[mid]==n) return mid;
            else if(arr2[mid]>n) end = mid-1;
            else if(arr2[mid]<n) start=mid+1;
        }
        return start;
    }
}