P1102 A-B 数对

传送门

这题卡数据卡的。
这里有两种方法，如果纯暴力的话是n^2，肯定超时.
所以第一种是二分nlogn，还有一种是桶排序A-B=C,我们求出A的数量*B的数量，B怎么求呢，B=A-C.

二分

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+10;
int n;
ll c,sum;
ll a[maxn];
map<ll,ll> q;
void binary(int l,int r,int x){
	int mid;
	while(l<=r){
		mid=l+(r-l)/2;
		if(x-a[mid]>c) l=mid+1;
		else if(x-a[mid]<c) r=mid-1;
		else{
			sum+=q[a[mid]];
			break;
		}
	}
}
int main(){
	scanf("%d%lld",&n,&c);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		q[a[i]]++;
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++){
		binary(1,n,a[i]);
	}
	printf("%lld\n",sum);
	return 0;
}

桶排序

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
map<ll,ll> q;
int n;
ll c,sum;
int main(){
	scanf("%lld%lld",&n,&c);
	for(int i=1;i<=n;i++)
	{
		ll x;
		scanf("%lld",&x);
		q[x]++;
	}
	for(map<ll,ll>::iterator it=q.begin();it!=q.end();it++){
		sum+=it->second*q[it->first-c];
	}
	printf("%lld\n",sum);
}