【CodeForces】895B. XK Segments 题解

题目来自于：CodeForces - 895B

首先按照题意：

第一行输入三个数字，分别是：
1.第二行需要输入多少个元素；
2.每两个元素所对应的区间要处以的数，判断是否等于0；
3.区间里的元素应该等于0的数量。

然后，第二行输入对应的元素。

我直接试了一下两个for，O(n^2)的复杂度 超时了。

所以这道题的关键点在于：理解题意，如何统计，以及降低时间复杂度。

这里我们可以先排序，然后再使用二分查找。

#include<bits/stdc++.h>
using namespace std;

int main(){
	long long n,x,k,i,l,r,isum=0,a[100005];
	cin>>n>>x>>k;
	for(i=0;i<n;i++)
		cin>>a[i];
	sort(a,a+n);//对数组进行排序
	for(i=0;i<n;i++){
		l=(a[i]-1)/x;//计算初始位置
		r=l+k;//求出满足k个元素的位置
		isum+=lower_bound(a,a+n,(r+1)*x)-lower_bound(a, a+n, max(r*x, a[i]));//利用二分法找出满足条件的最大值与最小值之差有多少个
	}
	cout<<isum;//差即是结果
}
//y SIUS, contest: Codeforces Round #448 (Div. 2), problem: (B) XK Segments, Accepted, # 

lower_bound( begin,end,num)可以返回一个有序序列的不小于num的值的地址，不存在则返回end。常用用法：通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。

关键思想：假设我们要取3的倍数，那么如果从8开始取，取三个倍数，那么首位应该就是(8-1)/3=2；2+3=5; 那么就会从8到(5+1)*3=18里面得到，数字就是9、12、15，根据数字计算就是18-max(15,8)=3，得到最终答案。