这道题好水,就是不好理解而已......
思路
暴力的复杂度是 n 2 n^2 n2 显然不可能通过
不难想到先排序,然后再使用二分查找。
lower_bound(begin, end, num)
可以返回一个有序序列 的不小于 n u m num num 的值的地址,不存在则返回 e n d end end。常用用法:通过返回的地址减去起始地址 b e g i n begin begin,得到找到数字在数组中的下标。
代码
cpp
#include <bits/stdc++.h>
#define int long long
using namespace std;
int n, x, k, a[100010], cnt;
int main() {
ios :: sync_with_stdio(false);
cin >> n >> x >> k;
for (int i = 1; i <= n; i ++) cin >> a[i];
sort(a + 1, a + n + 1);
for (int i = 1; i <= n; i ++) {
int l = (a[i] - 1) / x, r = l + k;
cnt += lower_bound(a + 1, a + n + 1, (r + 1) * x) - lower_bound(a + 1, a + n + 1, max(r * x, a[i]));
}
cout << cnt << endl;
return 0;
}