upper_bound()
lower_bound()
对于这两个函数的使用,对于每一个 v[i],查找第一个大于等于v[i]+m的位置,再查找到查找第一个小于等于v[i]+m的位置,二者相减,就是v[i]这个数完成对应要求 A-B=C 的个数,累加起来就好,注意开 long long 来完成题目,不开long long 见祖宗.
#include<bits/stdc++.h>
#define endl "\n";
using namespace std;
typedef long long ll;
const int N=1e6+5;
typedef pair<int,int> PII;
const int INF=0x3f3f3f3f;
typedef struct node{
}node;
void IOS(){
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
}
//试试upper_bound();
int main(){
IOS();
ll n,m;
cin>>n>>m;
vector<ll>v(n);
for(int i=0;i<n;++i){
cin>>v[i];
}
sort(v.begin(),v.end());
ll cnt=0;
for(int i=0;i<n;++i){
cnt+=(upper_bound(v.begin(),v.end(),v[i]+m)-v.begin())-(lower_bound(v.begin(),v.end(),v[i]+m)-v.begin());
}
cout<<cnt;
return 0;
}