计蒜客题解——T1562：二分查找（三）题解

题目相关

题目链接

计蒜客，https://nanti.jisuanke.com/t/T1562。

题目描述

蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了，所以蒜头君也不知道数组里面有什么数字，所以蒜头君会经常询问在数组 A 中，比 x 大的最小值是多大？但是这次蒜头君要求这个数字必须大于 x，不能等于 x。

输入格式

第一行输入两个整数 n 和 m，分别表示数组的长度和查询的次数。

接下来一行有 n 个整数 ai​。

接下来 m 行，每行有 1 个整数 x，表示蒜头君询问的整数。

输出格式

对于每次查询，如果可以找到，输出这个整数。

否则输出 −1。

样例输入

10 5
1 1 1 2 3 5 5 7 8 9
0
1
4
9
10

样例输出

1
2
5
-1
-1

数据范围

1 ≤ n, m ≤10^5, 0 ≤ x ≤ 10^6。

分析

标准的二分查找模板题。

题目要求

输入一个 x，在数组 A 中找到这个数字必须大于 x，不能等于 x。注意是数组中的数据，不是下标。

样例输入分析

输入数据为 [1 1 1 2 3 5 5 7 8 9]。

第一个输入 x=0，意味着找大于 0 的最小数据是多少？数组最小的数为 1，自然对应的数为 1，对应索引是 0，输出也就是 a[0]。

第二个输入 x=1，意味着找大于 1 的最小数据是多少？自然对应的最小值 2，对应索引是 3，输出也就是 a[3]。

第三个输入 x=4，意味着找大于 4 的最小数据是多少？数组没有数据 4，比 4 大的是 5，对应索引是 5，输出也就是 a[5]。

第四个输入 x=9，意味着找大于 9 的最小数据是多少？数组最大值为 9，因此输出 -1。

第五个输入 x=10，意味着找大于 10 的最小数据是多少？数组最大值为 9，因此输出 -1。

因此对应的样例输出为

1
2
5
-1
-1

通过上面的分析，我们可以发现，就是在数组 a 找出 x 的右边界。

编程思路

1、读入数组 A。

2、对 A 进行排序。

3、读入一个 x，在数组 a 找出 x 的右边界。

AC 参考代码

STL 版本

利用 algorithm 的 upper_bound 函数。

#include <cstdio>
#include <algorithm>

const int MAXN = 1e5+4;
int nums[MAXN];

int main() {
    int n,m;
    scanf("%d%d", &n, &m);
    for (int i=0; i<n; i++) {
        scanf("%d", &nums[i]);
    }

    //排序
    std::sort(nums, nums+n);

    for (int i=0; i<m; i++) {
        int x;
        scanf("%d", &x);

        if (x>=nums[n-1]) {
            printf("-1\n");
        } else if (x<nums[0]) {
            printf("%d\n", nums[0]);
        } else {
            int pos = std::upper_bound(nums, nums+n, x)-nums;
            printf("%d\n", nums[pos]);
        }
    }

    return 0;
}

C++ 版本

#include <cstdio>
#include <algorithm>

const int MAXN = 1e5+4;
int nums[MAXN];

int upper_bound(int *A, int l, int r, int val){ // [l, r)
    while (l<=r) {
        int mid = l + ((r-l)>>1);
        if(A[mid] <= val) {
            l = mid+1;
        } else {
            r = mid-1;
	}
    }
 
    return l;
}

int main() {
    int n,m;
    scanf("%d%d", &n, &m);
    for (int i=0; i<n; i++) {
        scanf("%d", &nums[i]);
    }

    //排序
    std::sort(nums, nums+n);

    for (int i=0; i<m; i++) {
        int x;
        scanf("%d", &x);

        if (x>=nums[n-1]) {
            printf("-1\n");
        } else if (x<nums[0]) {
            printf("%d\n", nums[0]);
        } else {
            int pos = upper_bound(nums, 0, n-1, x);
            printf("%d\n", nums[pos]);
        }
    }

    return 0;
}