CF1538C - Number of Pairs

最新推荐文章于 2022-01-07 14:53:39 发布

Zerotogether

最新推荐文章于 2022-01-07 14:53:39 发布

阅读量343

点赞数

分类专栏： CodeForces 题解文章标签： c++ 算法二分查找

本文链接：https://blog.csdn.net/Zerotogether/article/details/122333460

版权

题解同时被 2 个专栏收录

21 篇文章 0 订阅

订阅专栏

CodeForces

16 篇文章 0 订阅

订阅专栏

题目链接

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

题目描述

【题目描述】

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

【输入格式】

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

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

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

【输出格式】

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

解题思路

首先把数组从小到大排序。
对于第 $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 \geq l$ .也是使用二分查找；
遍历每个位置，即可统计出所有满足条件的所有数对。

参考代码

#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;
}