邻值查找【链表or set】

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_37867156/article/details/82224414

题目描述:给定一个长度为 n(n<=10^5)的序列 A,A 中的数各不相同。对于 A 中的每一个数 Ai,求:\small \min (1\leqslant j< i)\left | A_{i}-A_{j} \right | 以及令上式取到最小值的 j(记为 Pi)。若最小值点不唯一,则选择使 Aj 较小的那个。

分析:我们维护一个有序集合,则集合中与 Ai 最接近的值要么是 Ai 的前驱,要么是 Ai 的后继。比较前驱、后继与Ai的差即可。

#include <cstdio>
#include <iostream>
#include <set>
#define INF 0x3f
using namespace std;
struct node{
    int x,index;
    bool operator<(const node &a)const{
        return x<a.x;
    }
};
set<node> s;
int main(){
    int n,x;
    scanf("%d",&n);
    scanf("%d",&x);
    s.insert((node){x,1});
    for(int i=2;i<=n;i++){
        scanf("%d",&x);
        set<node>::iterator ir=s.lower_bound((node{x,0})),il=ir;
        il--;
        if(ir==s.end()){
            printf("%d %d\n",x-il->x,il->index);
        }else if(ir==s.begin()){
            printf("%d %d\n",ir->x-x,ir->index);
        }else if(x-il->x<=ir->x-x){
            printf("%d %d\n",x-il->x,il->index);
        }else{
            printf("%d %d\n",ir->x-x,ir->index);
        }
        s.insert((node{x,i}));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37867156/article/details/82224414
set