优先队列——A-B

题目描述

给出一串数以及一个数字C，要求计算出所有A-B=C的数对的个数。

（不同位置的数字一样的数对算不同的数对）

输入格式

第一行包括2个非负整数N和C，中间用空格隔开。

第二行有N个整数，中间用空格隔开，作为要求处理的那串数。

输出格式

输出一行，表示该串数中包含的所有满足A-B=C的数对的个数。

样例输入

4 1

1 1 2 3

样例输出

3

数据范围

对于90%的数据，N <= 2000；

对于100%的数据，N <= 200000。

所有输入数据都在int范围内。

思路

用两个优先队列(小根堆 相当于将数据从小到大排一遍序）

因为差是固定的 随意减数和被减数在两个队列中相对前后顺序是一样的

遍历一遍即可

代码（C++)

#include <cstdio>#include <queue>using namespace std;int n,c,m,k,num,w;char x;long long a=0,b=0,ans=0;priority_queue <int,vector<int>,greater<int> > p,q;inline int read();int main(){n=read();c=read();for(int i=1;i<=n;++i)m=read(),p.push(m),q.push(m);while(!p.empty()&&!q.empty()){k=p.top();a=0;b=0;while(p.top()==k)++a,p.pop();while(!q.empty()&&q.top()<c+k)q.pop();while(!q.empty()&&q.top()==c+k)++b,q.pop();ans+=a*b;}printf("%lld",ans);return 0;}inline int read(){num=0;x=getchar();while(!((x>=48&&x<=57)||x==45))x=getchar();if(x==45)w=-1,x=getchar();elsew=1;while(x>=48&&x<=57)num=num*10+x-48,x=getchar();return num*w;}