5G通讯（c++版）

题目描述

一条直线上有 $n$ 个点，第 $i$ 个点的坐标为 $x_i$。给定一个限制 $d$，若两点距离不超过 $d$，那么它们可以直接通讯。请统计有多少对点可以直接通讯。

输入描述

• 第一行，两个整数表示 $n$ 与 $d$。
• 第二行，$n$个整数，表示$x_1​,x_2​,…,x_n$​。

数据范围

• 对于30%的数据，$1≤n≤1000$；
• 对于70%的数据，$1≤n≤10,000$；
• 对于100%的数据，$1≤n≤100,000$，
• $1≤d≤1,000,000,000$，
• $1≤x_i​≤1,000,000,000$。

输出描述

单个整数：表示有多少对点可以直接通讯。

输入样例

4 25
30 40 10 20

输出样例

5

#include<iostream>
#include<algorithm>
using namespace std;
 
int a[100005];

int main() {
    int n, d;
    cin >> n >> d;
    for (int i = 0; i < n; ++i)
        cin >> a[i];
    sort(a, a+n);
    
    long long sum = 0;
    int j = 0;
    for (int i = 0; i < n; ++i) {
        while (j < n and a[j] - a[i] <= d) 
            j++;
            // 定义j的位置为 第一个超过d的位置 
            // j停下来的时候，有两种可能
            // 1 越界
            // 2 x[j] - x[i] > d 
        sum += j - i - 1;
    }
    cout << sum;
    
    return 0;
}