洛谷 P2249 【深基13.例1】查找 题解 二分查找

【深基13.例1】查找

题目描述

输入 $n$ 个不超过 $10^9$ 的单调不减的（就是后面的数字不小于前面的数字）非负整数 $a_1,a_2,\dots ,a_n$，然后进行 $m$ 次询问。对于每次询问，给出一个整数 $q$，要求输出这个数字在序列中第一次出现的编号，如果没有找到的话输出 $-1$ 。

输入格式

第一行 $2$ 个整数 $n$ 和 $m$，表示数字个数和询问次数。

第二行 $n$ 个整数，表示这些待查询的数字。

第三行 $m$ 个整数，表示询问这些数字的编号，从 $1$ 开始编号。

输出格式

输出一行，$m$ 个整数，以空格隔开，表示答案。

样例 #1

样例输入 #1

11 3
1 3 3 3 5 7 9 11 13 15 15
1 3 6

样例输出 #1

1 2 -1

提示

数据保证，$1\le n\le 10^6$，$0\le a_i,q\le 10^9$，$1\le m\le 10^5$

本题输入输出量较大，请使用较快的 IO 方式。

原题

洛谷P2249——传送门

做题背景

本蒟蒻于Apr/30/2024 23:55提交的二分查找代码被Hack。特来欺负板子题泄泄火。

思路

二分查找板子题，可采用 lower_bound 函数解决该问题。不过鉴于是板子题，我还是手敲了细致入微的二分代码。二分查找主要是基于分治思想，每次判断所求位于二分后的左区间还是右区间，实现详见代码。

代码

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

bool check(int a, int x) // 该处数组元素大于等于x则满足查找需求
{
    if (a >= x)
        return 1;
    else
        return 0;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int n, m;
    cin >> n >> m;
    vector<int> a(n); // 不递减数组
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    int x; // 询问的数字
    while (m--)
    {
        cin >> x;
        int l = 0, r = n - 1; // 上下界为数组索引的上下界
        // 二分查找数组中第一个大于等于x的位置
        while (l < r)
        {
            int mid = (l + r) / 2;
            if (check(a[mid], x)) // check函数自己写
            {
                r = mid; // 这里是 r = mid, 说明[l,mid]是合法范围
            }
            else
            {
                l = mid + 1; //  [l,mid]这个范围都不是合法范围，所以下一次查找直接从 l = mid + 1开始了
            }
        }
        // 最后得到的l和r相等，均为查找得到的位置，取其一即可
        if (a[l] == x) // 如果数组中存在数x，输出其位置
        {
            cout << l + 1 << ' '; // 由于数组下标从0开始，而题目要求从1开始，所以这里要加1
        }
        else // 如果不存在x，输出-1
        {
            cout << -1 << ' ';
        }
    }

    return 0;
}