Codeforces-895B XK Segments

• 题意:输入n,x,k; 和一个 n 长度的 序列 , 一个二元组的定义如下(i,j)
  
  • 满足 ai<= y <= aj, 使得y|x的y个数为k个(y为[ai,aj]闭区间的整数)
  • 注意 (i,j) != (j,i) 如果 i!=j的情况下,求出满足这样的二元组的个数
• 题解:
  
  • 因为仅仅是求得个数,先排序,
  • 枚举每一个元素,对每一元素ai,求得 y1>=ai 的满足y|x的第一个元素,y1 = l * x; l = ai/x+(ai%x !=0)
  • 如果 ai 能整除 x 则 y1 = x * (ai/x)
  • 否则 y1 = x * (ai/x +1)
  • eg x = 3 , ai = 3 y1 = 3;
  • ai = 4 y1 = 6;(y1是大于等于ai的第一个被x整除的数)
  • 于是要满足有k个,y1已经是第1个了，说明后面还得有k-1个,f1 = y1 + (k-1)*x;
  • f1 是满足 >= ai ,且 [ai,f1] 的闭区域内恰好有 k 个 y 满足 y|x 的最小数
  • 于是 上界为 f2 = f1 + x ; [ai,f2]的闭区域中有 k+1个 y 满足 y|x
  • 综上 对于 ai 能满足的区间为 [f1,f2);
  • 于是 ans+=lower_bound(a+1,a+n+1,f2)lower_bound(a+1,a+n+1,f1);
  • 注意到k = 0 的 时候要特判一下

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;

const int N=1e5+1e3+7;

long long n,k,x,a[N],ans;

int main()
{
    cin>>n>>x>>k;
    for(int i=1;i<=n;i++)
        scanf("%I64d",&a[i]);
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
    {
        long long l=a[i]/x+(a[i]%x!=0);
        long long f1=(l+k-1)*x,f2=f1+x;
        //cout<<f1<<" "<<f2<<endl;
        if(k==0)
        {
            f1=a[i];
            f2=l*x;
        }
        ans+=lower_bound(a+1,a+n+1,f2)-lower_bound(a+1,a+n+1,f1);
    }
    printf("%I64d\n",ans);
    return 0;
}