牛客——救救兔子(lower_bound()函数)

链接:https://ac.nowcoder.com/acm/contest/372/B
来源:牛客网
 

救救兔子

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld

题目描述

某天,一只可爱的小兔砸在路上蹦蹦跳跳地走着,怪人PM6出现了,于是小兔子被盯上了。

PM6:“免子。哦不,小兔子。你长得真好…不对,真可爱。我这里有一道很容易很容易的题目,答对了我就请你吃萝卜,答错了你就请我吃兔肉,好不好呀~~?”

小兔砸:“萝卜!?好呀好呀好呀。”于是笨笨的兔纸入套了。

PM6:“我这里有一个由 N 个数组成的序列,给你 M 个询问,每个询问会给你一个数 X ,对于每个询问,你要回答出序列中与这个值最接近的元素。”

听完题后,兔子吓成一坨免子了,面临着变成红烧兔头的危险,求求你救救兔子!

输入描述:

第一行包含一个整数N,为序列长度。
第二行包含N个整数,为序列各元素。
第三行包含一个整数M,为PM6的询问个数。
接下来M行,每行一个整数X,为要询问最接近元素的给定值。
对于40%的数据:1<=N<=10000,1<=M<=1000
对于另外10%的数据:M=1
对于100%的数据:1 <=N<= 100000,1<=M<=10000,0<=序列中的每个数,X<=1e9

输出描述:

M行,每行有一个整数,为最接近相应给定值的元素值,保持输入顺序。若有多个值满足条件,输出最小的一个。

示例1

输入

复制

5
2 4 5 5 7
3
2
5
6

输出

复制

扫描二维码关注公众号,回复: 5300589 查看本文章
2
5
5

思路很简单就是,用lower_bound ()函数找到比key值大的第一个元素,还要判断和比key值小的那个数的差值判断大小,但是当这个值为0 或者n 是没有可以判断的所以直接输出就行

lower_bound()函数,就是返回比key值大的第一个元素的位置

对应lower_bound()函数是upper_bound()函数,它返回比key值大的最后一个元素的位置

#include<iostream>
#include<algorithm>
using namespace std;
int a[100005];
int main(){
	int n,m;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	    scanf("%d",&a[i]);
	sort(a,a+n);
	scanf("%d",&m);
	while(m--){
		int x;
		scanf("%d",&x);
		int flag=lower_bound(a,a+n,x)-a;
		if(flag==0)
		    cout<<a[0]<<endl;
		else if(flag==n)
		    cout<<a[n-1]<<endl;
		else{
			if(abs(a[flag-1]-x)<=abs(a[flag]-x))
			    cout<<a[flag-1]<<endl;
			else
			    cout<<a[flag]<<endl;
		}
	}
	return 0;
} 

猜你喜欢

转载自blog.csdn.net/red_red_red/article/details/87901003