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

最近在准备二分的课件，讲真不想写这个题解。不是因为题目难，而是这个题目太简单，我竟然翻车了。第一次提交出现了 Timeout，实在太太丢人了。后面仔细阅读了题目，再三推敲，尼玛，坑竟然是给的数列不是有序的。重要的事情说三次，不是有序的，不是有序的，不是有序的。哎天哪，实在太惨不忍睹了。还是写一个题解报告吧。丢人就丢人，语文太差。

题目相关

题目链接

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

题目描述

蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了，所以蒜头君也不知道数组里面有什么数字，所以蒜头君会经常询问在数组 A 中，大于等于 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
1
5
9
-1

数据范围

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

分析

标准的二分查找模板题。本题的难度在于对语文的理解。

题目要求

输入一个 x，在数组 A 中找到大于等于 x 的最小值是多大。注意是数组中的数据，不是下标。靠，我摔倒的第二个地方。

样例输入分析

输入数据为 [1 1 1 2 3 5 5 7 8 9]，（这个数据太具有误导性了，加上二分查找必须在有序数列里进行，我直接掉坑里了）。

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

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

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

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

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

因此对应的样例输出为

1
1
5
9
-1

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

编程思路

1、读入数组 A。

2、对 A 进行排序。

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

AC 参考代码

STL 版本

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

const int MAXN = 1e5+6;
int a[MAXN] = {};

int main() {
    //读入数据
    int n,m;
    scanf("%d%d", &n, &m);
    int i;
    for (i=0; i<n; i++) {
        scanf("%d", &a[i]);
    }

    //排序
    sort(a, a+n);

    //查询数据
    for (i=0; i<m; i++) {
        int x;
        scanf("%d", &x);

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

    return 0;
}

注意：使用 cin 读入数据在计蒜客有 3 个测试用例是超时的。

C++ 版本

直接套用寻找左边界模板即可。

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

const int MAXN = 1e5+6;
int a[MAXN] = {};

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

int main() {
    //读入数据
    int n,m;
    scanf("%d%d", &n, &m);
    int i;
    for (i=0; i<n; i++) {
        scanf("%d", &a[i]);
    }

    //排序
    sort(a, a+n);

    //查询数据
    for (i=0; i<m; i++) {
        int x;
        scanf("%d", &x);

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

    return 0;
}