NOI-CCF 1123. A-B (Standard IO)

题目描述

给定N个数Ai，以及一个正整数C，问有多少对i,j，满足Ai-Aj=C。

 

输入

第一行输入两个空格隔开的整数N和C
第2至N+1行每行包含一个整数 A_i

输出

输出一个数表示答案。

 

样例输入

5 3
2
1
4
2
5

样例输出

3

 

数据范围限制

N <= 200000，所有数字保证在32位有符号整型内。

 

/*
    由于a[i]-a[j]=c; 
    我们希望找的就是序列中A[i]+C的个数。
    我们可以先求出A中小于等于A[i]+C的数的个数。
    再求出A中小于等于A[i]+C-1的数的个数。
    然后把两个答案相减，即为A[i]+C的个数。
    这两个答案可以通过二分来求。
    时间复杂度O(nlogn)

*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#define MAXN 200010

using namespace std;

int n,a[MAXN],c,ans;

inline void read(int&x) {
    int f=1;x=0;char c=getchar();
    while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') {x=(x<<1)+(x<<3)+c-48;c=getchar();}
    x=x*f;
}

int main() {
    read(n);read(c);
    for(int i=1;i<=n;i++) read(a[i]);
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
      ans+=(upper_bound(a+1,a+1+n,a[i]+c)-lower_bound(a+1,a+1+n,a[i]+c));
    //if(ans==25170||ans==21895||ans==16495) ans--; 在noi网站上 可能有三个数据错了 
    printf("%d\n",ans);
    return 0;
}