题目链接

https://www.luogu.com.cn/problem/P1296?contestId=34234
(来自洛谷)

解题思路

洛谷数据一直十分的水,本题也是。直接暴力也可以解出,但是蓝桥必然不会出现水题,因此要学会优质算法。
用STL库中的upper_bound函数。核心代码不过就是函数那句。

AC代码

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int N=1e6+5;
int n,d;
int a[N];
int main(){
    cin>>n>>d;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    sort(a+1,a+n+1);

    ll ans=0;
    for(int i=1;i<n;i++){
        int tmp=upper_bound(a+i+1,a+n+1,a[i]+d)-a-i-1;//在i牛之后,有多少牛可以与之交流
    //    cout<<tmp<<endl;//检验一下
        ans+=tmp;
    }
    cout<<ans;
}