codeforces 895B. XK Segments（二分）

最新推荐文章于 2025-03-20 20:51:52 发布

原创最新推荐文章于 2025-03-20 20:51:52 发布 · 371 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

【二分/尺取/差分】

本文介绍了一种解决区间内整除计数问题的方法。通过排序和二分查找技术，文章详细阐述了如何高效地计算出给定区间内能够被特定整数整除的数的数量。

题意：给你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;
}