力扣2563. 统计公平数对的数目
这一题的意思是让找数组中所有的公平对数,什么是所谓的公平对数呢,就是这两个数的和处在lower和upper之间的,就是符合的公平对数,那么怎么找呢?
首先我们可以对数组进行排序,因为公平数对并没有顺序限制只需要两个数的和处在某一个区间即可。当数组处于升序的时候,就可以采用二分的方式来寻找一个数处在一个范围之间。
我们可以固定一个数的位置,然后来判断与它匹配的另一个数是否在lower-固定的数和upper-固定的数。
如果在说明是符合公平对数的,我们只需要用二分来找到最早出现与固定的数匹配的和最晚出现位置,两者之差就是当固定一个数时,与这个数相匹配的所有数的个数。
完整代码如下:
class Solution {
public:long long countFairPairs(vector<int>& nums, int lower, int upper) {long long ans=0;sort(nums.begin(),nums.end());//可以对数组进行排序,因为我们不在意数对的先后顺序//那么从左到右对于每一个点,我们可以尝试在它的右边找一个nums[j]//使得这个nums[j]>=lower-nums[i]// nums[j]<=upper-nums[i]for(int i=0;i<nums.size();i++){int l=i+1;int r=nums.size()-1;int lx=INT_MAX;while(l<=r){int mid=(l+r)/2;if(nums[mid]>=lower-nums[i]&&nums[mid]<=upper-nums[i]){r=mid-1;lx=min(lx,mid);}else if(nums[mid]<lower-nums[i]){l=mid+1;}else{r=mid-1;}}l=i+1;r=nums.size()-1;int rx=-1;while(l<=r){int mid=(l+r)/2;if(nums[mid]>=lower-nums[i]&&nums[mid]<=upper-nums[i]){l=mid+1;rx=max(rx,mid);}else if(nums[mid]<lower-nums[i]){l=mid+1;}else{r=mid-1;}}if(rx==-1||lx==INT_MAX){ans+=0;}else{ans+=rx-lx+1;}}return ans;}
};
时间复杂度O(nlogn)