P1102 A-B 数对

最新推荐文章于 2024-09-13 16:27:15 发布

代码无bug

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量372

点赞数

分类专栏：二分文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/aaronyyds/article/details/130569448

版权

二分专栏收录该内容

3 篇文章 0 订阅

订阅专栏

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n,c;
	cin>>n>>c;
	vector<int>p;
	p.push_back(-1);
	for(int i=1;i<=n;i++){
		int a;
		cin>>a;
		p.push_back(a);
	}
	long long ans=0;
	sort(p.begin(),p.end());
	for(int i=1;i<=n;i++){
		int cnt1=0,cnt2=0;
		int k=p[i];
		int l=i+1,r=n;
		while(l<r){
			int mid=(l+r)/2;  //靠左 
			if(p[mid]-k>=c)r=mid;
			else l=mid+1;
		}
		if(p[l]-k==c)cnt1=l;
		else continue;
		l=cnt1-1;
		r=n;
		while(l<r){
			int mid=(l+r+1)/2;// 靠右 
			if(p[mid]<=p[cnt1])l=mid;
			else r=mid-1;
		}
		ans+=r-cnt1+1;
	}
	cout<<ans;
}