计蒜客-1562-二分查找

蒜头君手上有个长度为 nn 的数组 AA。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问在数组 AA 中,比 xx 大的最小值是多大?但是这次蒜头君要求这个数字必须大于 xx,不能等于 xx。

输入格式
第一行输入两个整数 nn 和 mm,分别表示数组的长度和查询的次数。

接下来一行有 nn 个整数 a_ia
i

接下来 mm 行,每行有 11 个整数 xx,表示蒜头君询问的整数。

输出格式
对于每次查询,如果可以找到,输出这个整数。

否则输出 -1−1。

数据范围
1 \le n, m \le 10^5, 0 \le x \le 10^61≤n,m≤10
5
,0≤x≤10
6

输出时每行末尾的多余空格,不影响答案正确性

样例输入复制
10 5
1 1 1 2 3 5 5 7 8 9
0
1
4
9
10
样例输出复制
1
2
5
-1
-1

本题两个地方可以进行优化,1:如果x小于a[0],直接输出a[0],如果x>=a[n-1],直接输出-1, 2:用scanf和printf优化cin 和cout

//@author:hairu,wu
//@from:ahut

#include
#include
using namespace std;

int n,m;
int a[1000000];


int main(){
	cin >> n>> m;
	for(int i=0;i<n;i++){
		cin >> a[i];
	}
	sort(a,a+n);	
	for(int i=0;i<m;i++){
		int x;
		scanf("%d",&x);
		if(x<a[0] ){
			printf("%d\n",a[0]);
			continue;
		}
		
		if(x>=a[n-1]){
			printf("-1\n");
			continue;
		}
		
		int ans=-1;
		int left=0,right=n-1;
		while(left<=right){
			int mid=(left+right)>>1;
			if(a[mid]>x){
				ans=a[mid];
				right=mid-1;
			}else{
				left=mid+1;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}

你可能感兴趣的:(二分)