题目描述
输入 nn 个不超过 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 方式。
解题思路
本题的核心在于高效地处理大量数据的查询。由于给定了一个单调不减的整数序列和一系列查询,我们可以通过二分查找的方法来快速定位每个查询值在序列中第一次出现的位置。二分查找在有序数组中非常有效,能够将搜索时间复杂度降低到O(log n)。
对于每个查询,我们从序列的中间开始比较,如果中间值大于等于查询值,则说明查询值可能在左侧(包括中间位置),我们缩小搜索范围到左半部分;如果中间值小于查询值,则说明查询值一定在右侧,我们更新搜索范围为右半部分。当搜索范围缩小到只剩一个元素时,如果这个元素等于查询值,那么它就是我们要找的结果;如果不等,则说明查询值不在序列中。
示例代码
#include<bits/stdc++.h>
using namespace std;
int n,m,q,a[1000005];
int find(int x) //二分查找
{
int l=1,r=n;
while (l<r)
{
int mid=l+(r-l)/2;
if (a[mid]>=x) r=mid;
else l=mid+1;
}
if (a[l]==x) return l;
else return -1;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n; i++)
{
scanf("%d",&a[i]);
}
for (int i=1 ; i<=m ; i++)
{
scanf("%d",&q);
int ans=find(q);
printf("%d", ans);
if (i < m) printf(" "); // 注意,在最后一个元素后不加空格
}
printf("\n"); // 输出结束后换行
return 0;
}