二分查找与二分答案——【深基13.例1】查找

二分查找与二分答案——【深基13.例1】查找

题目描述

输入 n(n ≤ 10⁶) 个不超过 10⁹ 的单调不减的（就是后面的数字不小于前面的数字）非负整数 a₁,a₂,…,a_n，然后进行 m(m ≤ 10⁵) 次询问。对于每次询问，给出一个整数 q(q ≤ 10⁹)，要求输出这个数字在序列中的编号，如果没有找到的话输出 -1。

输入格式

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

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

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

输出格式

m 个整数表示答案。

输入输出样例

输入样例

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

输出样例

1 2 -1

分析

因为输入的 n 个数字本身就是有序的，所以可以直接使用 lower_bound() 来找，lower_bound(begin,end,x) 用于查找第一个大于或等于 x 的位置，不存在则返回 end。

#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[1000005];

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