您的位置:首页 > 大数据 > 人工智能

hdu 5178 pairs (二分)

2015-03-26 15:58 316 查看
求序列x中满足a<b且x[b]-x[a]的绝对值不超过K的数对(a,b)的个数。

排序后二分查找每个数加上k后的上界pos,结果累加pos-i即可

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
#define LL __int64
int x[100005];
int main()
{
    int T,n,k;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        int i;
        LL ans=0;
        for(i=0;i<n;++i) scanf("%d",&x[i]);
        sort(x,x+n);
        for(i=0;i<n;++i)
        {
           int pos=upper_bound(x+i,x+n,x[i]+k)-x;
           --pos;
           ans+=(LL)(pos-i);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: