给出n个数,位于x轴上,求满足 | num[a] - num[b] | <=k 的对数。先排序,然后直接线性扫描。
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
typedef __int64 ll;
const int maxn=100005;
int a[maxn];
int n;
int main()
{
int k,T;
scanf("%d",&T);
while(T--)
{
scanf("%d %d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
sort(a+1,a+1+n);
ll ans=0;
int p=1;
for(int i=1;i<=n;i++)
{
while(a[p]-a[i]<=k&&p<=n)p++;
ans+=p-i-1;
}
printf("%I64d\n",ans);
}
return 0;
}