4923: [Lydsy六月月赛]K小值查询 splay

Description
维护一个长度为n的正整数序列a_1,a_2,…,a_n，支持以下两种操作：
1 k，将序列a从小到大排序，输出a_k的值。
2 k，将所有严格大于k的数a_i减去k。

题解：

很巧妙啊，利用操作次数有限这个条件对部分数进行暴力操作。然后这道题好像写一个点只有一个数的好写一点，然后就有一些地方不同，注意一下，经常splay，就行了。还顺便学了一下指针的splay，但是好像非指针快一点，这里就放非指针的代码了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010;
const int inf=2100000000;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int son[Maxn][2],fa[Maxn],size[Maxn];
int v[Maxn],tag[Maxn];
int root=0,tot=0;
void up(int x)
{
    if(!x)return;
    int lc=son[x][0],rc=son[x][1];size[x]=1;
    if(lc)size[x]+=size[lc];
    if(rc)size[x]+=size[rc];
}
void work(int x,int t){v[x]-=t,tag[x]+=t;}
void down(int x)
{
    if(!x)return;
    int lc=son[x][0],rc=son[x][1],t=tag[x];
    if(lc)work(lc,t);
    if(rc)work(rc,t);
    tag[x]=0;
}
void rotate(int x)
{
    int y=fa[x],z=fa[y],w=(son[y][0]==x);
    son[y][w^1]=son[x][w];if(son[x][w])fa[son[x][w]]=y;
    son[z][son[z][1]==y]=x;fa[x]=z;
    son[x][w]=y;fa[y]=x;
    up(y),up(x);
}
int sta[Maxn],top;
void update(int x)
{
    top=0;
    while(1){if(!x)break;sta[++top]=x;x=fa[x];}
    while(top)down(sta[top--]);
}
void splay(int x,int rt)
{
    if(!x)return;
    update(x);
    while(fa[x]!=rt)
    {
        int y=fa[x],z=fa[y];
        if(z==rt)rotate(x);
        else rotate(((son[y][1]==x)==(son[z][1]==y))?y:x),rotate(x);
    }
    if(!rt)root=x;
}
int Find(int x)
{
    int t=root;
    while(v[t]!=x)
    {
        if(tag[t])down(t);
        int tmp=son[t][x>v[t]];
        if(tmp)t=tmp;
        else break;
    }return t;
}
int find_kth(int k)
{
    int t=root;
    while(1)
    {
        if(tag[t])down(t);
        int lc=son[t][0],rc=son[t][1];
        if(k==size[lc]+1)
        {
            splay(t,0);
            break;
        }
        if(k<=size[lc])t=lc;
        else t=rc,k-=(size[lc]+1);
    }return v[t];
}
int find_pre(int k)
{
    int x=root,re;
    while(x)
    {
        if(tag[x])down(x);
        if(v[x]<k)re=x,x=son[x][1];
        else x=son[x][0];
    }return re;
}
int find_next(int k)
{
    int x=root,re;
    while(x)
    {
        if(tag[x])down(x);
        if(v[x]>k)re=x,x=son[x][0];
        else x=son[x][1];
    }return re;
}
void add(int x,int f)
{
    int t=++tot;v[t]=x;tag[t]=0;size[t]=1;
    son[f][x>v[f]]=t;fa[t]=f;
    up(f);splay(f,0);
}
void ins(int x)
{
    if(!root){root=1;add(x,0);return;}
    int f=Find(x);
    while(1)
    {
        if(tag[f])down(f);
        int tmp=son[f][x>v[f]];
        if(!tmp)break;
        else f=tmp;
    }
    add(x,f);
}
int tl[Maxn],len;
void dfs(int x)
{
    tl[++len]=x;
    if(tag[x])down(x);
    int lc=son[x][0],rc=son[x][1];
    if(lc)dfs(lc);
    if(rc)dfs(rc);
}
void Work(int k)
{
    int L,R;len=0;
    /*[k+1,2k]*/
    L=find_pre(k+1),R=find_next(k<<1);
    splay(L,0);splay(R,L);
    if(son[R][0])dfs(son[R][0]),fa[son[R][0]]=0,son[R][0]=0,up(R);
    for(int i=1;i<=len;i++)
    {
        int t=tl[i];
        v[t]-=k;tag[t]=0;son[t][0]=son[t][1]=0;size[t]=1;
        int f=Find(v[t]);
        while(1)
        {
            if(tag[f])down(f);
            int tmp=son[f][v[t]>v[f]];
            if(tmp)f=tmp;
            else
            {
                son[f][v[t]>v[f]]=t;
                fa[t]=f;
                up(f),splay(f,0);
                break;
            }
        }
    }
    /*[2k+1,inf]*/
    L=find_pre((k<<1)+1),R=2;
    splay(L,0);splay(R,L);
    if(son[R][0])work(son[R][0],k);
}
int n,m;
int main()
{
    n=read(),m=read();
    ins(0),ins(inf+10);
    for(int i=1;i<=n;i++)ins(read());
    while(m--)
    {
        int op=read(),k=read();
        if(op==1)printf("%d\n",find_kth(k+1));
        else Work(k);
    }
}