hdu 5178 pairs

pairs

问题描述

John 在X轴上拥有nn个点，他们的坐标分别为$(x[i],0),(i=0,1,2,…,n-1)$。 他想知道有多少对< a,b ><a,b>满足|x[b]-x[a]| \leq k(a < b)∣x[b]−x[a]∣≤k(a<b)。

输入描述

第一行包含一个正整数TT(大约5)，表示有多少组数据。
对于每一组数据，先读入两个数n,k(1 \leq n \leq 100000,1 \leq k \leq {10}^{9})n,k(1≤n≤100000,1≤k≤10​9​​)。
接下来nn行，分别输入x[i]({-10}^{9} \leq x[i] \leq {10}^{9},x[i]x[i](−10​9​​≤x[i]≤10​9​​,x[i]为整数)。

输出描述

对于每组数据，输出一行表示有多少对< a,b ><a,b>满足|x[b]-x[a]| \leq k∣x[b]−x[a]∣≤k。

输入样例

2
5 5
-100
0
100
101
102
5 300
-100
0
100
101
102

输出样例

3
10

很显然不能使用两重循环，先排序，对于每一个坐标，向右找到最右的而且满足条件的位置，那么这点的答案为两点之间的所有点，对于找点，二分可得

 

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long LL;
const int Max=+;
int a[Max];
int main()
{
    int T;
    for(scanf("%d",&T);T;T--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        for(int i=;i<=n;i++) scanf("%d",&a[i]);
        sort(a+,a++n);
        LL ans=;
        for(int i=;i<=n;i++)
        {
            int l=i+,r=n;
            while(l<=r)
            {
                int mid=(l+r)>>;
                if(abs(a[i]-a[mid])<=k) l=mid+;
                else  r=mid-;
            }
            ans+=l--i;
        }
        printf("%I64d\n",ans);
    }
    return ;
}