洛谷P1102 A-B 数对

最新推荐文章于 2025-12-29 22:37:36 发布

原创最新推荐文章于 2025-12-29 22:37:36 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

该C++代码示例中，文章主要展示了如何使用upper_bound()和lower_bound()函数在排序数组中查找特定值的上下界，并通过计算这两个边界之间的差值，求解一个关于数组元素满足特定条件(A-B=C)的数量问题。

upper_bound()
lower_bound()

对于这两个函数的使用，对于每一个 v[i],查找第一个大于等于v[i]+m的位置，再查找到查找第一个小于等于v[i]+m的位置，二者相减，就是v[i]这个数完成对应要求 A-B=C 的个数，累加起来就好，注意开 long long 来完成题目，不开long long 见祖宗.

#include<bits/stdc++.h>
#define endl "\n";
using namespace std;
typedef long long ll;
const int N=1e6+5;
typedef pair<int,int> PII;
const int INF=0x3f3f3f3f;
typedef struct node{

}node;
void IOS(){
	 ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
}
//试试upper_bound(); 
int main(){
	IOS();
	ll n,m;
	cin>>n>>m;
	vector<ll>v(n);
	for(int i=0;i<n;++i){
		cin>>v[i];
	}
	sort(v.begin(),v.end());
	ll cnt=0;
	for(int i=0;i<n;++i){
		cnt+=(upper_bound(v.begin(),v.end(),v[i]+m)-v.begin())-(lower_bound(v.begin(),v.end(),v[i]+m)-v.begin());
	}
	cout<<cnt;
	return 0;
}