YTU 3734 查找最接近的元素 二分

查找最接近的元素

时间限制: 1.00s | 内存限制: 128MB

题目描述

在一个非降序列中，查找与给定值最接近的元素。

输入

第一行包含一个整数 n，为非降序列长度。1≤n≤100000

第二行包含 nnn 个整数，为非降序列各元素。所有元素的大小均在 0−10000000000 -之间。

第三行包含一个整数 m，为要询问的给定值个数。1≤m≤10000。

接下来 m 行，每行一个整数，为要询问最接近元素的给定值。所有给定值的大小均在 0−10000000000之间。

输出

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

输入输出样例

样例输入 #1

复制

3
2 5 8
2
10
5

样例输出 #1

复制

8
5

解题思路：

这道题要找和给定数最接近的数，那么就利用二分的两个模板。
模板1：找第一个大于等于x的数字，并返回其下标
模板2：找最后一个小于等于x的数字，并返回其下标

下面的代码解释了这两个模板 https://www.acwing.com/problem/content/791/
对应了数的范围这道题。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N];
int SL(int l,int r,int x)  // 寻找第一个大于等于给定数的数字，返回其下标
{
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(a[mid]>=x) r=mid;
        else l=mid+1;
    }
    return l;
}
int SR(int l,int r,int x) // 寻找最后一个小于等于给定数的数字，返回其下标
{
    while(l<r)
    {
        int mid=(l+r+1)>>1;
        if(a[mid]<=x) l=mid;
        else r=mid-1;
    }
    return r;
}
int main()
{
    int n,q;
    scanf("%d %d",&n,&q);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    while(q--)
    {
        int x; scanf("%d",&x);
        if(a[SL(0,n-1,x)]!=x) printf("-1 -1\n");
        else
        {
            printf("%d %d\n",SL(0,n-1,x),SR(0,n-1,x));
        }
    }
    return 0;
}

知道了二分的两个模板，那么这道题就好办了
找到第一个大于等于x 的数value1，找到最后一个小于等于x 的数value2，看看value1 和 value2 哪一个和x的绝对值的差值小 如何value1 和 value2 和x 一样接近，我们取value2
如果我们找不到大于等于x 的数value的话，求得的下标l 其实是最后一个小于等于x 的下标
比如：
3  
2 5 8
输入10 ，我们找第一个大于等于10 的数字的下标，没有，返回的l下标为3，是最后一个小于等于10的下标，所以不妨碍我们寻找距离x 最近的数。
 同样的如果我们找不到小于等于x 的数，返回的是第一个大于等于x的数字的下标
比如如果输入是1 ，没有第一个小于等于1 的数字,返回的是数字1 ，是第一个大于等于1的数字的下标。

AC代码 ：

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
int a[N];
int main()
{
    int n;
    cin>>n;
    for(int i=1; i<=n; i++) cin>>a[i];
    int m;
    cin>>m;
    while(m--)
    {
        int x;
        cin>>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;
        }
        int value1=a[l];
        l=1,r=n;
        while(l<r)
        {
            int mid=(l+r+1)/2;
            if(a[mid]<=x) l=mid;
            else r=mid-1;
        }
        int value2=a[l];
        if(abs(value1-x)<abs(value2-x)) cout<<value1<<endl;
        else if(abs(value1-x)>abs(value2-x)) cout<<value2<<endl;
        else
        {
            cout<<value2<<endl;
        }
    }
    return 0;
}