6355.统计公平数对的数目
题目
解题过程
1.这道题主要运用了两次二分查找,二分查找前我们需要先对数组进行排序
2.由条件可知,nums[i] + nums[j] >= lower, 那么在nums[i]已知的情况下, 我们可以得到 nums[j] >= lower - nums[i];
同理nums[i] + nums[j] <= upper, 那么在nums[i]已知的情况下, 我们可以得到 nums[j] <= upper - nums[i];
如果存在至少一个有效的配对, 最后我们可以得到一个范围(X, Y) X <= Y, 在这个区间内的任何一个坐标与i结合都符合要求.
3.我们以每一个坐标为i, 寻找后面的有效的j
4.首先nums[j]一定大于等于nums[i]的, 因为数组现在是非降序的
利用二分查找(left = i + 1, right = n - 1), 寻找nums[i] + nums[X] >= lower, X距离i的最近的坐标
如果不存在, 那么就跳过, 因为两个坐标都必须为有效的
然后利用二分查找(left = i + 1, right = n - 1), 寻找nums[i] + nums[Y] <= upper, Y距离i的最远的坐标
如果不存在, 那么就跳过 因为两个坐标都必须为有效的
5.那么与当前坐标i匹配的数对的数量就为index2 - index1 + 1;
6.统计每一个坐标作为i的有效数对数量和即可
具体实现步骤以及二分法的补充
代码
class Solution {
public long countFairPairs(int[] nums, int lower, int upper) {
long result = 0;
int n = nums.length;
Arrays.sort(nums);
for(int i = 0; i < n - 1; i++){
int index1 = binary_search1(nums, lower - nums[i], i + 1);
if(index1 == -1){
continue;
}
int index2 = binary_search2(nums, upper - nums[i], i + 1);
if(index2 == -1){
continue;
}
result += index2 - index1 + 1;
}
return result;
}
public int binary_search1(int[] nums, int target, int index){
int left = index;
int right = nums.length - 1;
while(left < right){
int mid = (left + right) / 2;
if(nums[mid] >= target){
right = mid;
}
else{
left = mid + 1;
}
}
return nums[right] >= target ? right : -1;
}
public int binary_search2(int[] nums, int target, int index){
int left = index;
int right = nums.length - 1;
while(left < right){
int mid = (left + right + 1) / 2;
if(nums[mid] <= target){
left = mid;
}
else{
right = mid - 1;
}
}
return nums[right] <= target ? right : -1;
}
}