LeetCode 1385题解析：二分查找优化数组距离计算

今晚摘大星星吗

1. 问题理解与解法概述

今天我们来拆解LeetCode第1385题"两个数组间的距离值"。这道题看似简单，但要想写出高效的解法，需要我们对二分查找有深入的理解。题目要求我们找出arr1中满足特定条件的元素个数：对于arr1中的每个元素，arr2中所有元素与该元素的绝对差都必须大于给定的d值。

1.1 问题重述

给定两个整数数组arr1和arr2，以及一个整数d，我们需要计算arr1中满足以下条件的元素数量：对于arr1中的元素x，arr2中所有元素y都满足|x - y| > d。换句话说，arr1中的元素x与arr2中最近的元素的距离也要大于d。

1.2 暴力解法思路

最直观的解法是暴力枚举：对于arr1中的每个元素，遍历arr2中的所有元素，检查是否所有绝对差都大于d。这种解法的时间复杂度是O(n*m)，其中n和m分别是arr1和arr2的长度。当数组较大时（比如都达到1000），这种解法会达到百万次操作，效率较低。

1.3 优化思路

我们可以通过预处理arr2来优化算法。具体思路是：

先对arr2进行排序（O(m log m)）
对于arr1中的每个元素x，在排序后的arr2中使用二分查找找到最接近x的元素
计算这两个元素的绝对差，如果这个最小差值大于d，则计数加一

这种优化后的算法时间复杂度为O(m log m + n log m)，当m和n较大时，效率提升明显。

2. 二分查找解法详解

2.1 排序预处理

首先我们需要对arr2进行排序。C++中可以直接使用sort函数：

cpp复制sort(arr2.begin(), arr2.end());

排序后，arr2中的元素按升序排列，这样我们就可以使用二分查找来快速定位元素。

2.2 二分查找实现

对于arr1中的每个元素x，我们需要在arr2中找到：

第一个大于等于x的元素（lower_bound）
这个元素的前一个元素（如果存在）

这两个元素是arr2中距离x最近的元素，我们只需要比较这两个元素与x的差值即可。

cpp复制ind = lower_bound(arr2.begin(), arr2.end(), arr1[i]) - arr2.begin();

2.3 边界情况处理

二分查找时需要注意边界情况：

如果lower_bound返回的是arr2.end()，说明arr2中所有元素都小于x，此时只需比较x与arr2最后一个元素的差值
如果lower_bound返回的是arr2.begin()，说明arr2中所有元素都大于等于x，此时只需比较x与arr2第一个元素的差值
一般情况，比较x与找到的元素及其前一个元素的差值

2.4 差值计算与判断

计算x与找到的最近元素的绝对差，并判断是否大于d：

cpp复制if(ind == m) mi = abs(arr1[i] - arr2[m-1]);
else {
    mi = abs(arr1[i] - arr2[ind]);
    if(ind - 1 >= 0) mi = min( mi, abs(arr1[i] - arr2[ind-1]) );
}
if(mi > d) ret++;

3. 完整代码解析

让我们完整分析一下给出的代码：

cpp复制class Solution {
public:
    int findTheDistanceValue(vector<int>& arr1, vector<int>& arr2, int d) {
        sort(arr2.begin(), arr2.end());  // 排序arr2
        int n = arr1.size(), m = arr2.size(), ind, mi, ret = 0;
        
        for(int i = 0; i < n; i++) {
            // 找到arr2中第一个大于等于arr1[i]的元素位置
            ind = lower_bound(arr2.begin(), arr2.end(), arr1[i]) - arr2.begin();
            
            // 处理边界情况并计算最小差值
            if(ind == m) mi = abs(arr1[i] - arr2[m-1]);
            else {
                mi = abs(arr1[i] - arr2[ind]);
                if(ind - 1 >= 0) mi = min( mi, abs(arr1[i] - arr2[ind-1]) );
            }
            
            // 如果最小差值大于d，计数加一
            if(mi > d) ret++;
        }
        return ret;
    }
};