A-B数对-CSDN博客

本文链接：https://blog.csdn.net/wsdsbtzk/article/details/147098815

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

题目描述

给出一串正整数数列以及一个正整数 C，要求计算出所有满足 A−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 N,C。

第二行，N 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 A−B=C 的数对的个数。

输入输出样例

输入 #1复制

4 1
1 1 2 3

输出 #1复制

说明/提示

对于 75% 的数据，1≤N≤2000。

对于 100% 的数据，1≤N≤2×105，0≤ai<230，1≤C<230。

2017/4/29 新添数据两组

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int a[N];
int l,r;
int n,c;
long long int res=0;
int find1(int x)
{
	l=-1;
	r=n;
	while(l+1!=r)
	{
		int mid=(l+r)>>1;
		if(a[mid]<x)
		{
			l=mid; 
		}
		else
		{
			r=mid;
		}
	}
	if(a[r]==x)
	{
		return r;
	}
	else
	{
		return -1;
	}
}
int find2(int x)
{
	l=-1;
	r=n;
	while(l+1!=r)
	{
		int mid=(l+r)>>1;
		if(a[mid]<=x)
		{
			l=mid;
		}
		else
		{
			r=mid;
		}
	}
	if(a[l]==x)
	{
		return l;
	}
	else
	{
		return -1;
	}
}
int main ()
{
	cin>>n>>c;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		
	}
	sort(a,a+n);
	for(int i=0;i<n;i++)
	{
		int x=a[i]+c;
		if(find1(x)!=-1)
		{
			res+=find2(x)-find1(x)+1;
		}
	}
	cout<<res<<endl;
	return 0;
}
//by crtzk7