洛谷&小码王 (P1102&C2L11) [A+B数对]

【A-B数对】

题目描述

给出N个数以及一个数字 C，要求计算出所有 A - B = C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行两个整数 N, C。

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

输出格式

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

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示

2-1=1

2-1=1

3-2=1

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

对于100%的数据，1≤N≤2*10^5,保证所有输入数据都在 int 范围内。

上——————代——————码

#include <bits/stdc++.h>
using namespace std;
int a[1000000]={0};
int sumerize=0;
int main(){
    int n,m;
    cin >> n >> m;
    for(int i=0; i<n; i++){
        cin >> a[i];
    }
    sort(a,a+n);
    for(int i=0; i<n; i++){
        sumerize+=((upper_bound(a,a+n,a[i]+m)-a)-(lower_bound(a,a+n,a[i]+m)-a));
    }
    cout << sumerize;
    return 0;
}