【Leetcode 每日一题】2563. 统计公平数对的数目-CSDN博客

本文链接：https://blog.csdn.net/2401_88859777/article/details/147343416

问题背景

给你一个下标从 $0$ 开始、长度为 $n$ 的整数数组 $n u m s$ ，和两个整数 $l o w er$ 和 $u pp er$ ，返回公平数对的数目。
如果 $(i, j)$ 数对满足以下情况，则认为它是一个公平数对：

$\le i < j < n$ ，且
$lower≤nums[i]+nums[j]≤upperlower\le nums[i] + nums[j] \le upper$

数据约束

$\le nums.length \le 10 ^ 5$
$n u m s . l e n g t h = n$
$\le nums[i] \le 10 ^ 9$
$\le lower \le upper \le 10 ^ 9$

解题过程

对于任意一个 $n u m s [j]$ ，能与它构成公平数对的 $n u m s [i]$ 范围是 $\le nums[j] \le upper - nums[i]$ 。
由于题目要求只是从数组中挑选两个数字，数组中元素的顺序不影响最终的结果，可以先排序，用二分查找。

具体实现

class Solution {
    public long countFairPairs(int[] nums, int lower, int upper) {
        Arrays.sort(nums);
        long res = 0;
        for (int i = 0; i < nums.length; i++) {
            int right = binarySearch(nums, i, upper - nums[i] + 1);
            int left = binarySearch(nums, i, lower - nums[i]);
            res += right - left;
        }
        return res;
    }

    private int binarySearch(int[] nums, int right, int target) {
        int left = 0;
        while (left < right) {
            int mid = left + ((right - left) >>> 1);
            if (nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid;
            }
        }
        return left;
    }
}