upperbound的使用

Description

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

Format

Input

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

Output

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

Samples

【输入样例】

输入数据 1

6 
1 2 2 2 3 4
3
2
5
3

Copy

输出数据 1

5
no
6

Copy

【输出样例】

Limitation

1s, 1024KiB for each test case.

思路

直接做，代码见下：

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int main(){
	int a[1099048];
	map<long long,long long>::iterator bak;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	a[n+1]=1e9+8;
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>k;
		int pos2=upper_bound(a+1,a+n+n,k)-a; 
		if(pos2<=n) cout<<pos2<<endl;
		else cout<<"NO"<<endl;
	}
	return 0;
}