BlackBox

标签：define   插入   排列   iostream   output   read   body   break   NPU   

Description
囧囧有个囧盒，可以处理一些很囧的事情。这个囧盒一开始是空的，可以处理2种操作：
Add(X)：将元素X放入囧盒。
Get：第I次调用Get的时候，输出囧盒里面第I小的数。下面是囧囧操作的一个例子：

LT看到囧囧这个囧盒很羡慕，他想搞个类似的家伙来，于是找到了你。

Input
第一行两个数N，M（1<=N，M<=30000），分别表示ADD操作的个数和Get操作的个数。
第二行N个MaxLongint范围内的整数，表示ADD操作顺序插入的元素值。
第三行M个数（按非降序排列），表示M条Get命令的位置（位于第几条ADD命令之后）

Output
M行，分别对应M条GET操作的输出。（任意时刻Get次数不会超过当前Add次数）

Sample Input
7 4
3 1 -4 2 8 -1000 2
1 2 6 6

Sample Output
3
3
1
2

splay板子题，直接找第k大即可。两个堆的写法更快

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline int read(){
    int x=0,f=1;char ch=getchar();
    for (;ch<‘0‘||ch>‘9‘;ch=getchar())  if (ch==‘-‘)    f=-1;
    for (;ch>=‘0‘&&ch<=‘9‘;ch=getchar())    x=(x<<1)+(x<<3)+ch-‘0‘;
    return x*f;
}
inline void print(int x){
    if (x>=10)     print(x/10);
    putchar(x%10+‘0‘);
}
const int N=3e4;
struct Splay{
    #define T(x) (tree[f[x]][1]==x)
    int tree[N+10][2],f[N+10],size[N+10],val[N+10];
    int root,len;
    void updata(int x){size[x]=size[tree[x][0]]+size[tree[x][1]]+1;}
    void move(int x){
        int fa=f[x],son=tree[x][T(x)^1];
        tree[x][T(x)^1]=fa;
        tree[fa][T(x)]=son;
        if (son)    f[son]=fa;
        f[x]=f[fa];
        if (f[x])   tree[f[x]][T(fa)]=x;
        f[fa]=x;
        updata(fa),updata(x);
    }
    void splay(int x){
        while (f[x]){
            if (f[f[x]])    T(x)==T(f[x])?move(f[x]):move(x);
            move(x);
        }
        root=x;
    }
    void insert(int x){
        val[++len]=x;
        if (!root){
            size[root=len]=1;
            return;
        }
        int i=root;
        while (1){
            size[i]++;
            if (x<=val[i]){
                if (!tree[i][0]){f[tree[i][0]=len]=i;break;}
                i=tree[i][0];
            }else{
                if (!tree[i][1]){f[tree[i][1]=len]=i;break;}
                i=tree[i][1];
            }
        }
        splay(len);
    }
    int Get_pre(){
        int x=tree[root][0];
        while (tree[x][1])  x=tree[x][1];
        return x;
    }
    int find(int x,int i){
        if (size[tree[i][0]]+1==x)  return i;
        if (x<=size[tree[i][0]])    return find(x,tree[i][0]);
        return find(x-size[tree[i][0]]-1,tree[i][1]);
    }
}T;
int A[N+10],Get[N+10];
int main(){
    int n=read(),m=read();
    for (int i=1;i<=n;i++)  A[i]=read();
    for (int i=1;i<=m;i++)  Get[read()]++;
    for (int i=1,k=0;i<=n;i++){
        T.insert(A[i]);
        while (Get[i]--)    printf("%d\n",T.val[T.find(++k,T.root)]);
    }
    return 0;
}

BlackBox

标签：define   插入   排列   iostream   output   read   body   break   NPU   

原文地址：https://www.cnblogs.com/Wolfycz/p/9478960.html