CF1538C - Number of Pairs

题目链接

https://codeforces.com/problemset/problem/1538/C

题目描述

【题目描述】

给出一个由整数组成的数组 $a$，求一对整数 $(i,j)$（$1\le i\le j\le n$）满足 $l\le a_i+a_j\le r$ 的数量。

【输入格式】

在输入的第一行为一个整数 $t$（$1\le t\le {10}^4$），为数据组数。

接下来对于每组数据，第一行为三个整数 $n,l,r$（$1\le n\le 2\times {10}^5$，$1\le l\le r\le {10}^9$），为数组的长度和上文中的 $l,r$。第二行有 $n$ 个整数 $a_1,a_2,\dots ,a_n$（$1\le a_i\le {10}^9$）表示数组 $a$。

保证对于所有组数据 $\sum n\le 2\times {10}^5$。

【输出格式】

对于每组数据，输出满足条件的 $(i,j)$ 组数。

解题思路

1. 首先把数组从小到大排序。
2. 对于第$i$个数，在序列$a_1$到$a_{i-1}$找到最大的一个数$a_k$使得$a_k+a_i\le r$.很明显，使用二分查找即可；同样的，在序列$a_1$到$a_{i-1}$找到最小的一个数$a_k$使得$a_k+a_i\ge l$.也是使用二分查找；
3. 遍历每个位置，即可统计出所有满足条件的所有数对。

参考代码

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int MAXN=200005;
int a[MAXN];
int t,n,l,r;
int  SearchLeft(int left, int right, int x){
	int res=0, mid;
	while( left <= right ) {
		mid=(left+right)>>1;
		if( a[mid]+a[x] >= l ) {
			res=mid;
			right=mid-1;
		}
		else left=mid+1;
	}
	return res;
}

int  SearchRight(int left, int right, int x){
	int res=0, mid;
	while( left <= right ) {
		mid=(left+right)>>1;
		if( a[mid]+a[x] <= r ) {
			res=mid;
			left=mid+1;
		}
		else right=mid-1;
	}
	return res;
}
int main(){
	scanf("%d", &t);
	while( t-- ) {
		int left,right;
		long long  ans=0;
		scanf("%d%d%d", &n, &l, &r);
		for(int i=1; i<=n; i++) 
			scanf("%d", &a[i]);
		sort(a+1,a+1+n);
		for(int i=n; i>=2; i--) {
			left=SearchLeft(1,i-1,i);
			right=SearchRight(1,i-1,i);	//	printf("left:%d  right:%d\n",left,right);
			if( left==0 || right==0 ) 
				continue;//break;是错的 
			ans += right-left+1;
		}
		printf("%lld\n",ans);
		
	}
	return 0;
}