P2249 【深基13.例1】查找

链接：P2249 【深基13.例1】查找 - 洛谷
 

题目描述

输入 n 个不超过 109 的单调不减的（就是后面的数字不小于前面的数字）非负整数 a1​,a2​,…,an​，然后进行 m 次询问。对于每次询问，给出一个整数 q，要求输出这个数字在序列中第一次出现的编号，如果没有找到的话输出 −1 。

输入格式

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

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

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

输出格式

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

输入输出样例

输入 #1复制运行

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

输出 #1复制运行

1 2 -1 

说明/提示

数据保证，1≤n≤106，0≤ai​,q≤109，1≤m≤105

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

方法一：二分

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

const int N=1e6+5;
int a[N];
int n,m;

void find(int x){
    int l=1;
    int r=n;
    while(l<r){
        int mid=(l+r)/2;
        if(a[mid]>=x){
            r=mid;
        }else{
            l=mid+1;
        }
    }
    if(a[l]!=x){
        cout<<"-1 ";
    }else{
        cout<<l<<" ";
    }
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    while(m--){
        int x;
        cin>>x;
        find(x);
    }     
    return 0;
}

方法二：STL

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

const int N=1e6+5;

int n,m;
int a[N];

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    while(m--){
        int x;
        cin>>x;
        int ans=lower_bound(a+1,a+1+n,x)-a;
        if(a[ans]!=x){
            cout<<"-1 ";
        }else{
            cout<<ans<<" ";
        }
    }
    return 0;
}