【深基13.例1】查找
题目描述
输入 n n n 个不超过 1 0 9 10^9 109 的单调不减的(就是后面的数字不小于前面的数字)非负整数 a 1 , a 2 , … , a n a_1,a_2,\dots,a_{n} a1,a2,…,an,然后进行 m m m 次询问。对于每次询问,给出一个整数 q q q,要求输出这个数字在序列中第一次出现的编号,如果没有找到的话输出 − 1 -1 −1 。
输入格式
第一行 2 2 2 个整数 n n n 和 m m m,表示数字个数和询问次数。
第二行 n n n 个整数,表示这些待查询的数字。
第三行 m m m 个整数,表示询问这些数字的编号,从 1 1 1 开始编号。
输出格式
输出一行, m m m 个整数,以空格隔开,表示答案。
样例 #1
样例输入 #1
11 3
1 3 3 3 5 7 9 11 13 15 15
1 3 6
样例输出 #1
1 2 -1
提示
数据保证, 1 ≤ n ≤ 1 0 6 1 \leq n \leq 10^6 1≤n≤106, 0 ≤ a i , q ≤ 1 0 9 0 \leq a_i,q \leq 10^9 0≤ai,q≤109, 1 ≤ m ≤ 1 0 5 1 \leq m \leq 10^5 1≤m≤105
本题输入输出量较大,请使用较快的 IO 方式。
原题
做题背景
本蒟蒻于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;
}