P2249 【深基13.例1】查找

P2249 【深基13.例1】查找

题目描述

输入 n(n\le10^6)n(n≤106) 个不超过 10^9109 的单调不减的（就是后面的数字不小于前面的数字）非负整数 a_1,a_2,\dots,a_{n}a1,a2,…,a**n，然后进行 m(m\le10^5)m(m≤105) 次询问。对于每次询问，给出一个整数 q(q\le10^9)q(q≤109)，要求输出这个数字在序列中第一次出现的编号，如果没有找到的话输出 -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 

代码

#include<iostream>
using namespace std;
int n, m, ele;
int p[1000000];
int lower_bound(int A[], int left, int right, int ans)
{
	int mid;
	while (left < right)
	{
		mid = (left + right) / 2;
		if (A[mid] >= ans)
			right = mid;
		else
			left = mid + 1;
	}
	if (A[left] != ans)
		return -2;
	return left;
}
int main()
{
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> p[i];
	for (int i = 0; i < m; i++)
	{
		cin >> ele;
		cout << lower_bound(p, 0, n, ele) + 1 << " ";
	}
}