c++ 二分答案(数组查找)

数组查找2

题目描述

给定一个已经从小到大排好序的数组，数组内有n个元素：a[1]、a[2]……a[n]。现在，有一个数x，你需要在数组中找到一个i，使a[i]>x，并输出i（如果有多个i，输出最小的一个），如果不存在对应的i，输出“no”。

输入

输入包括多行：
第一行包含一个整数n（1<=n<10^7）；
第二行输入n个整数(每个数不会超过10^7)；
第三行输入一个整数k（1<=k<=1000），表示需要进行k次查找；
接下来输入k行，每行输入一个整数x(每个数不会超过10^7)

输出

输出k行，对应每次查找的结果。

样例输入

6 
1 2 2 2 3 4
3
2
5
3

样例输出

5
no
6

AC代码

#include<bits/stdc++.h>
using namespace std;
int a[10000001],n,t;
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    scanf("%d",&t);
    for (int i=1;i<=t;i++)
    {
        int key;
        scanf("%d",&key);
        int l=1,r=n;
        int res=-1;
        while (l<=r)
        {           
            int m=(l+r)/2;
            if (key>a[m]) l=m+1;
            else if(key<a[m])
            {
                res=m;
                r=m-1;
            }
            else if(key==a[m]) l=m+1;
        }
        if(res==-1) printf("no
");
        else printf("%d
",res);
    }
    return 0;
}