题意:给你N个数字,问有多少对L,R(L <= R)使得[L,R]的区间中恰好含有K个整除X的数?
思路:先排序,对于a[i],[1,a[i])中含有temp = (a[i] - 1) / X个被整除X的数,那么满足要求的右端点的区间为[X * (temp + k),X * (temp + k + 1))。然后枚举左端点二分查找满足要求的右端点区间。然后统计答案就可以了。有两个个细节需要注意
①对数可能会很多,要用long long
②我们二分求解时候每次都应该在整个区间去找。因为例如第三个样例中①[3,3]第一个3和第五个3 ②[3,3]第五个3和第一个3是两种不同的情况,对于当前枚举的左端点i,满足要求的右端点不一定在i的右边。但是这样又可能会多算上下面这种情况X = 10,K = 0,0 11 13 13 15 16 17 30这种情况我们会多加[17,15],[15,11]等这种区间,所以我们得想办法避免这种问题。假设通过二分求得的区间为[pos1,pos2),我们再求一个等于a[i]的第一个位置pos3。[max(pos1,pos3),pos2)就是满足要求的正确区间。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 5;
long long n,x,k,a[MAXN];
int main()
{
while(~scanf("%lld%lld%lld",&n,&x,&k))
{
for(int i = 0; i < n; i++) scanf("%lld",&a[i]);
sort(a,a + n);
long long ans = 0;
for(int i = 0; i < n; i++)
{
long long temp = (a[i] - 1) / x;
int pos1 = lower_bound(a, a + n, x * (temp + k)) - a;
int pos2 = lower_bound(a, a + n, x * (temp + k + 1)) - a;
int pos3 = lower_bound(a, a + n, a[i]) - a;
ans += (pos2 - max(pos1, pos3));
}
printf("%lld\n",ans);
}
return 0;
}