給你一個下標從?0?開始、長度為?n
?的整數數組?nums
?,和兩個整數?lower
?和?upper
?,返回?公平數對的數目?。
如果?(i, j)
?數對滿足以下情況,則認為它是一個?公平數對?:
0 <= i < j < n
,且lower <= nums[i] + nums[j] <= upper
示例 1:
輸入:nums = [0,1,7,4,4,5], lower = 3, upper = 6 輸出:6 解釋:共計 6 個公平數對:(0,3)、(0,4)、(0,5)、(1,3)、(1,4) 和 (1,5) 。
示例 2:
輸入:nums = [1,7,9,2,5], lower = 11, upper = 11 輸出:1 解釋:只有單個公平數對:(2,3) 。
提示:
1 <= nums.length <= 10^5
nums.length == n
-10^9?<= nums[i] <= 10^9
-10^9?<= lower <= upper <= 10^9
分析:先進行排序后,遍歷數組。對于每一個 nums[i],可以使用二分查找找到一個區間 [l,r],使得所有的 i∈[l,r] 滿足 lower?nums[j]≤nums[i]≤upper?nums[j]。具體來說,可以找到 ≤upper?nums[j] 的元素個數,減去 <lower?nums[j] 的元素個數,加入答案。
int find_index(int *nums,int numsSize,int target)
{int l=0,r=numsSize,m;while(l<r){int m=(l+r)/2;if(nums[m]>=target)r=m;else if(nums[m]<target)l=m+1;}return l;
}int cmp(const void *a,const void *b)
{return *(int*)a-*(int*)b;
}long long countFairPairs(int* nums, int numsSize, int lower, int upper) {long long ans=0;qsort(nums,numsSize,sizeof(int),cmp);for(int i=0;i<numsSize;++i){int l=find_index(nums,i,lower-nums[i]);int r=find_index(nums,i,upper-nums[i]+1);ans+=r-l;}return ans;
}