Hdu 5178 BestCoder #31 1001 Pairs 解题报告

题意简单,找到线段上距离小于k的点对的个数。

解法为,先排序所有点,然后遍历每个点x[a],二分刚好小于b=x[a]+k的所在点下标,表示a和a+1~b之间的点都为符合点。然后累加b-a即可。

用lower_bound会WA,要用upper_bound或者自己手写二分,问题出在当x[b]=x[a]+k时的区别。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
#define LL long long
#define MAXN 100005
int x[MAXN];
int main()
{
int T;
cin>>T;
while(T--)
{
int n,k;
scanf("%d%d",&n,&k);
for(int i = 0 ; i < n ; i++)
{
scanf("%d",&x[i]);
}
sort(x,x+n);
LL ans = 0;
for(int i = 0; i < n; i++)
{
int b = upper_bound(x + i + 1, x + n, x[i] + k) - x - i - 1;
// cout<<b<<endl;
ans += b;
}
cout<<ans<<endl;
}
}