A-B 数对

题目背景

出题是一件痛苦的事情！

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

题目描述

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

输入格式

输入共两行。

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

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

输出格式

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

样例 #1

样例输入 #1

1
2
4 1
1 1 2 3

样例输出 #1

1
3

提示

对于 $75%$ 的数据，$1 \leq N \leq 2000$。

对于 $100%$ 的数据，$1 \leq N \leq 2 \times 10^5$，$0 \leq a_i <2^{30}$，$1 \leq C < 2^{30}$。

2017/4/29 新添数据两组

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
 
typedef long long LL;
 
const int N = 2e5+10;
 
int f[N];
 
int main(){
    int n, c;
    cin>>n>>c;
    for(int i=0;i<n;i++) scanf("%d",&f[i]);
  
    LL ans = 0;
    sort(f,f+n,greater<int>());
  
    for(int i=0;i<n-1;i++){
        int x = f[i]-c;
        if(x<0) break;
        int l = i, r = n-1;
        while(l<r){
            int mid = l + r >> 1;
            if(f[mid]<=x) r = mid;
            else l = mid+1;
        }
        if(f[l]==x){
            ans++;
            int ll = l;
            r = n-1;
            while(ll<r){
                int mid = ll+r+1>>1;
                if(f[mid]>=x) ll = mid;
                else r = mid - 1;
            }
            ans+=r-l;
        }
    }
  
    cout<<ans<<endl;
  
  
    return 0;
}