[bzoj4923]K小值查询

来自FallDream的博客，未经允许，请勿转载，谢谢。

---

维护一个长度为n的正整数序列a_1,a_2,...,a_n，支持以下两种操作：

1 k，将序列a从小到大排序，输出a_k的值。

2 k，将所有严格大于k的数a_i减去k。

n,m<=100000

 

splay维护序列，每次操作小等于k的数字不变，大于2k的数字顺序不变，(k,2k]的数字减小至少一半，最多减小log次，暴力拿出来重新插入就行了。

复杂度nlog^2n

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define Rint register int 
#define ll long long
#define MN 100000
#define INF 1000000000000000000LL
#define getchar() (*S++) 
char B[1<<26],*S=B;
using namespace std;
inline 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 * 10 + ch - '0';ch = getchar();}
    return x * f;
}
int c[MN+5][2],fa[MN+5],rt=0,tag[MN+5],n,m,size[MN+5],q[MN+5],top=0;
ll a[MN+5];
inline void update(int x){size[x]=size[c[x][0]]+size[c[x][1]]+1;}
inline void Mark(int x,int v){tag[x]+=v;a[x]+=v;}
inline void pushdown(int x){Mark(c[x][0],tag[x]);Mark(c[x][1],tag[x]);tag[x]=0;}
 
void rotate(int x,int&k)
{
    int y=fa[x],z=fa[y],l=c[y][1]==x,r=l^1;
    if(y==k) k=x; else c[z][c[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
    c[y][l]=c[x][r];c[x][r]=y;
    update(y);update(x);
}
 
void Splay(int x,int&k)
{
    for(;x!=k;rotate(x,k))  
        if(fa[x]!=k) rotate((c[fa[fa[x]]][1]==fa[x]^c[fa[x]][1]==x)?x:fa[x],k);
}
 
void build(int&x,int l,int r,int last)
{
    if(l>r) {x=0;return;}x=l+r>>1;fa[x]=last;
    build(c[x][0],l,x-1,x);
    build(c[x][1],x+1,r,x);
    update(x);
}
 
int Ask_After(int x,int v)
{
    if(!x) return 0;int q;
    if(tag[x]) pushdown(x);
    if(a[x]>v) return (q=Ask_After(c[x][0],v))?q:x;
    else return Ask_After(c[x][1],v);
}
 
int Ask_Before(int x,int v)
{
    if(!x) return 0;int q;
    if(tag[x]) pushdown(x);
    if(a[x]<=v) return (q=Ask_Before(c[x][1],v))?q:x;
    else return Ask_Before(c[x][0],v); 
}
 
int Find(int x,int rk)
{
    int Sz=size[c[x][0]]+1;
    if(tag[x]) pushdown(x);
    if(Sz==rk) return x;
    if(Sz<rk) return Find(c[x][1],rk-Sz);
    else return Find(c[x][0],rk);
}
 
void Dfs(int x)
{
    if(tag[x]) pushdown(x);
    if(c[x][0]) Dfs(c[x][0]);      
    q[++top]=x;
    if(c[x][1]) Dfs(c[x][1]);
}
 
void Ins(int&x,int k,int last)
{
    if(!x){x=k;c[k][0]=c[k][1]=0;fa[k]=last;size[k]=1;return;} 
    if(tag[x]) pushdown(x);
    Ins(c[x][a[k]>=a[x]],k,x);
    update(x);
} 
 
int main()
{
    fread(B,1,1<<26,stdin);
    n=read();m=read();
    for(Rint i=1;i<=n;++i) a[i+1]=read();a[1]=-INF;a[n+2]=INF;
    sort(a+2,a+n+2);build(rt,1,n+2,0);
    for(Rint i=1;i<=m;++i)
    {
        int op=read(),v;
        if(op==1){printf("%d\n",(v=Find(rt,read()+1),Splay(v,rt),a[v]));}
        if(op==2)
        {
            v=read();int x=Ask_Before(rt,v),y=Ask_After(rt,v<<1); 
            Splay(x,rt);Splay(y,c[rt][1]);int z=c[y][0];
            if(z) {top=0;Dfs(z);fa[z]=0;c[y][0]=0;} 
            update(y);update(x);Mark(y,-v);
            if(z) for(Rint j=1;j<=top;++j) a[q[j]]-=v,Ins(rt,q[j],0),Splay(q[j],rt);
        }
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/FallDream/p/bzoj4923.html