[洛谷]P2249 查找

[洛谷]P2249 查找

题目描述

输入 $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 方式。

标签 ：二分

思路：

这题是二分的模板题，但是二分最容易错的地方在于while里的l和r的关系，到底是l<=r,还是l<r，要判断好，不然就会出错导致答案错误或者死循环

注意题目的要求是在序列中第一次出现的编号,相当于要往偏左的方向查找，我们可以理解范围在[l,r)的左闭右开的区间里，即l<r

#include<bits/stdc++.h>
using namespace std;
int n, m; const int N = 1e6 + 5;
int a[N], b;
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
    }
    for (int i = 0; i < m; i++)
    {
        scanf("%d", &b);
        int l = 1; int r = n;

        //cout << mid << " ";
        while (l < r)//在[l,r)的区间内查找。最终停止时是l==r
        {
            int mid = l + ((r - l) >> 1);//防越界写法，位运算加快速度
            if (a[mid] > b)r =mid;//中间值是在大于目标值的，说明我们可以舍弃右边的数字，mid可以这样想，a[mid]的值是大于b的，最终答案不是mid,但我们的右边界是开区间，因此我们取到mid
            else if (a[mid] == b)r = mid;
            else l = mid + 1;//中间值是大于目标值的我们把左半边给舍弃掉，l=mid+1
        }
        if (a[r] == b)printf("%d ", r);
        else printf("-1 ");
    }

    return 0;
}

第二种理解就是看成是[l,r]的左闭右闭区间，这样就可以写成while(l<=r)

#include<bits/stdc++.h>
using namespace std;
int n, m; 
int main()
{
    cin >> n >> m;
    vector<int>a(n);
    for (int i = 0; i < n; i++)cin >> a[i];
    while (m > 0)
    {
        int t;
        cin >> t;
        int l = 0,r = n - 1;
        while (l <= r) {
            int mid = (l + r) >> 1;
            if (a[mid] >= t)r = mid - 1;//因为要取到偏左的数字，我们可以把mid及其后面的数字舍弃，r=mid-1
            //为什么不写mid,因为在l<=r,当l==r的时候，且r=mid，这样就会死循环
            else l = mid + 1;
        }
        if (l == n)cout << -1 << " ";
        else {
            if (a[l] == t)cout << l + 1 << " ";
            else cout << -1 << " ";
        }
        m--;
    }
    return 0;
}