COJ0986：WZJ的数据结构（负十四）

写一个二分+树状数组+Treap，挺爽的

#include<cstdio>
#include<cctype>
#include<queue>
#include<ctime>
#include<cstring>
#include<algorithm>
#define rep(s,t) for(int i=s;i<=t;i++)
#define ren for(int i=first[x];i!=-1;i=next[i])
using namespace std;
inline int read() {
    int x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c==‘-‘) f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-‘0‘;
    return x*f;
}
const int maxn=100010;
const int maxnode=2000010;
struct Node {
    Node* ch[2];
    int r,s,v;
    void maintain() {s=ch[0]->s+ch[1]->s+1;}
}nodes[maxnode],*null=&nodes[0];
queue<Node*> del;
int n,A[maxn],ToT;
Node* newnode(int v) {
    Node* o;
    if(!del.empty()) o=del.front(),del.pop();
    else o=&nodes[++ToT];
    o->ch[0]=o->ch[1]=null;
    o->s=1;o->v=v;o->r=rand();
    return o;
}
void remove(Node* &o) {
    del.push(o);o=null;
}
void rotate(Node* &o,int d) {
    Node* k=o->ch[d^1];o->ch[d^1]=k->ch[d];k->ch[d]=o;
    o->maintain();k->maintain();o=k;
}
void insert(Node* &o,int v) {
    if(o==null) o=newnode(v);
    else {
        int d=v>o->v;insert(o->ch[d],v);
        if(o->ch[d]->r>o->r) rotate(o,d^1);
        else o->maintain();
    }
}
void remove(Node* &o,int v) {
    if(o->v==v) {
        Node* t=o;
        if(o->ch[0]==null) o=o->ch[1],remove(t);
        else if(o->ch[1]==null) o=o->ch[0],remove(t);
        else {
            int d=o->ch[0]->r>o->ch[1]->r;
            rotate(o,d);remove(o->ch[d],v);
        }
    }
    else remove(o->ch[v>o->v],v);
    if(o!=null) o->maintain();
}
void print(Node* o) {
    if(o==null) return;
    print(o->ch[0]);
    printf("%d ",o->v);
    print(o->ch[1]);
}
int query(Node* &o,int v) {
    if(o==null) return 0;
    if(v<=o->v) return query(o->ch[0],v);
    return query(o->ch[1],v)+o->ch[0]->s+1;
}
Node* root[maxn];
void insert(int x,int v) {for(;x<=n;x+=x&-x) insert(root[x],v);}
void update(int x,int v) {for(;x<=n;x+=x&-x) remove(root[x],v);}
int query(int x,int v) {int ret=0;for(;x;x-=x&-x) ret+=query(root[x],v+1);return ret;}
int main() {
    null->s=0;srand(time(0));
    n=read();int m=read();
    rep(1,n) root[i]=null;
    rep(1,n) insert(i,A[i]=read());
    while(m--) {
        if(read()) {
            int l=read(),r=read(),k=read();
            int L=1,R=100000,M;
            while(L<R) if(query(r,M=L+R>>1)-query(l-1,M=L+R>>1)>=k) R=M; else L=M+1;
            printf("%d\n",L);
        }
        else {
            int x=read();update(x,A[x]);
            insert(x,A[x]=read());
        }
    }
    return 0;
}

复习一下树状数组+可持久化线段树，写WA了一发真不爽

#include<cstdio>
#include<cctype>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(s,t) for(int i=s;i<=t;i++)
#define ren for(int i=first[x];i!=-1;i=next[i])
using namespace std;
inline int read() {
    int x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c==‘-‘) f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-‘0‘;
    return x*f;
}
const int maxn=100010;
const int maxnode=20000010;
int ls[maxnode],rs[maxnode],s[maxnode],ToT;
int n,m,A[maxn],root[maxn],c[maxn];
void update(int& y,int x,int l,int r,int pos,int v) {
    s[y=++ToT]=s[x]+v;if(l==r) return;
    int mid=l+r>>1;ls[y]=ls[x];rs[y]=rs[x];
    if(pos<=mid) update(ls[y],ls[x],l,mid,pos,v);
    else update(rs[y],rs[x],mid+1,r,pos,v);
}
void update(int x,int v) {
    for(int i=x;i<=n;i+=i&-i) update(c[i],c[i],1,100000,A[x],-1);
    for(int i=x;i<=n;i+=i&-i) update(c[i],c[i],1,100000,A[x]=v,1);
}
int lt[maxn],rt[maxn],ltot,rtot;
void get(int x,int tp) {
    if(!tp) {lt[ltot=1]=root[x];for(;x;x-=x&-x) if(c[x]) lt[++ltot]=c[x];}
    else {rt[rtot=1]=root[x];for(;x;x-=x&-x) if(c[x]) rt[++rtot]=c[x];}
}
int main() {
    n=read();m=read();
    rep(1,n) update(root[i],root[i-1],1,100000,A[i]=read(),1);
    while(m--) {
        if(!read()) {
            int x=read(),v=read();
            update(x,v);
        }
        else {
            int ql=read(),qr=read(),k=read();
            get(ql-1,0);get(qr,1);int l=1,r=100000;
            while(l<r) {
                int mid=l+r>>1,tot=0;
                rep(1,ltot) tot-=s[ls[lt[i]]];
                rep(1,rtot) tot+=s[ls[rt[i]]];
                if(tot>=k) {
                    r=mid;
                    rep(1,ltot) lt[i]=ls[lt[i]];
                    rep(1,rtot) rt[i]=ls[rt[i]];
                }
                else {
                    l=mid+1;k-=tot;
                    rep(1,ltot) lt[i]=rs[lt[i]];
                    rep(1,rtot) rt[i]=rs[rt[i]];
                }
            }
            printf("%d\n",l);
        }
    }
    return 0;
}