题解 -优美数对

题目描述

赵琳是爱学习的好孩子，虽然暑假出来探险，玩的很开心，但是她还是没有忘记自己的暑假作业。她把暑假作业也随身带在身上，暑假作业里有很多有趣的问题。例如，有一个由n个数组成的数组，a[1]到a[n],需要找出有多少个不同的数对(i,j)，满足a[i]<=a[j],并且恰好存在k个y，使得a[i]<=y<=a[j],y必须要被x整除。
赵琳觉得这个问题实在太难了，只能用计算机解决。于是她请求强子哥（光头强）帮他解决，可怜的光头强只能屁颠屁颠的去研究程序和算法了。在绞尽脑子之后，他终于做出来了，那你呢？
注意，这里i和j可以相等。

输入

输入第一行是3个整数n，x，k，n表示数组中元素的个数，x和k的含义见题目描述。
接下来一行有n个正整数，表示n个数。

输出

输出只有一行一个非负整数，表示有多少对(i,j)可以满足上面的条件。

样例输入

【样例1】
4 2 1
1 3 5 7
【样例2】
4 2 0
5 3 1 7
【样例3】
5 3 1
3 3 3 3 3

样例输出

【样例1】
3
【样例2】
4
【样例3】
25

提示

样例1解释：数对(1,2),(2,3),(3,4)都满足条件
样例2解释：数对(1,1),(2,2),(3,3),(4,4)都满足条件
样例3解释：任意两个数的数对都满足条件。

对于40%的数据，保证1<=n<=100，1<=a[i]<=10000,1<=x<=100,0<=k<=100。
对于70%的数据，保证1<=n<=1000，1<=a[i]<=1000000,1<=x<=1000,0<=k<=1000。
对于100%的数据，保证1<=n<=100000,1<=x<=1000000000,0<=k<=1000000000,
1<=a[i]<=1000000000。

思路

样例分析

对于a[i] = 1，x = 3，k = 2 的情况，

若想满足条件，结束区间必须是7，8

前置知识

[l,r] 中能被x整除的数的个数为 floor(r / x) - ceil(l / x) + 1

解析

我们可以枚举i，对于a[i]，找到满足题意的结束区间，区间中元素的个数即为符合题意的解的个数

由floor(a[j] / x) - ceil(a[i] / x) + 1 == k 得，floor(a[j] / x) = k - 1 + ceil(a[i] / x) ;

故 a[j] 所在的范围即为结束区间，设结束区间的起点和终点分别为l，r，

起点为 a[j] 整除 x 时，即 l = (k - 1 + ceil((double)a[i] / x)) * x，

终点为 r = l + x - 1。

由于已经排好序，故可以遍历a[i] ，用二分来查找原数组中 a[j] 的结束区间。

l，r对应的下标k1,k2，则k2 - k1 + 1即为合法解的个数

k1 = lower_bound(a + 1,a + n + 1,max(l,a[i])) - a，表示第一个大于等于max(l,a[i])的数的下标

k2 = upper_bound(a + 1,a + n + 1,r) - a - 1，表示第一个小于等于r的数的下标

代码

#include<bits/stdc++.h>
  
using namespace std;
  
typedef long long LL;
  
const int N = 100000 + 10;
  
LL n,x,k;
LL a[N];
LL res;
  
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
  
    cin >> n >> x >> k;
    for(int i = 1;i <= n;i++) cin >> a[i];
  
    sort(a + 1,a + 1 + n);
  
    for(int i = 1;i <= n;i++){
        LL l = (k - 1 + ceil((double)a[i] / x)) * x,r = l + x - 1;
        int k1 = lower_bound(a + 1,a + n + 1,max(l,a[i])) - a,k2 = upper_bound(a + 1,a + n + 1,r) - a - 1;
 
        res += k2 - k1 + 1;
    }
  
    cout << res;
  
    return 0;
}