bzoj 1901 Zju2112 Dynamic Rankings(动态区间第k大,主席树)

kk
O(nlognlogn)
O(nlogn+mlognlogn)

lognlogn


#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           10010
#define   MAXN          1000005
#define   maxnode       5
#define   sigma_size    30
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-6;
const LL     mod   = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

int a[MAX];
int b[MAX*2];
struct Node{
    int l,r,c;
}tr[MAX*200];
int root[MAX];
int bit[MAX];
int use[MAX];
int sz;

struct que{
    int op,l,r,k;
}p[MAX];

int build(int l,int r){
    int k=sz++;
    if(l==r){
        tr[k].c=0;
        return k;
    }
    int mid=(l+r)/2;
    tr[k].l=build(l,mid);
    tr[k].r=build(mid+1,r);
    return k;
}

int update(int rt,int l,int r,int pos,int d){
    int k=sz++;
    tr[k].c=tr[rt].c+d;
    if(l==r) return k;
    int mid=(l+r)/2;
    if(pos<=mid){
        tr[k].r=tr[rt].r;
        tr[k].l=update(tr[rt].l,l,mid,pos,d);
    }
    else{
        tr[k].l=tr[rt].l;
        tr[k].r=update(tr[rt].r,mid+1,r,pos,d);
    }
    return k;
}

int sum(int x){
    int ret=0;
    while(x){
        ret+=tr[tr[use[x]].l].c;
        x-=lowbit(x);
    }
    return ret;
}
int query(int L,int R,int lr,int rr,int l,int r,int k){
    if(l==r) return l;
    int mid=(l+r)/2;
    int tmp=sum(R)-sum(L)+tr[tr[rr].l].c-tr[tr[lr].l].c;
    if(tmp>=k){
        for(int i=L;i;i-=lowbit(i)) use[i]=tr[use[i]].l;
        for(int i=R;i;i-=lowbit(i)) use[i]=tr[use[i]].l;
        return query(L,R,tr[lr].l,tr[rr].l,l,mid,k);
    }
    else{
        for(int i=L;i;i-=lowbit(i)) use[i]=tr[use[i]].r;
        for(int i=R;i;i-=lowbit(i)) use[i]=tr[use[i]].r;
        return query(L,R,tr[lr].r,tr[rr].r,mid+1,r,k-tmp);
    }
}

int n;
int tot;
void modify(int x,int pos,int d){
    while(x<=n){
        bit[x]=update(bit[x],1,tot,pos,d);
        x+=lowbit(x);
    }
}

int main(){
    //freopen("in.txt","r",stdin);
        int q;
        scanf("%d%d",&n,&q);
        tot=0;
        for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[tot++]=a[i];
        sz=0;
        char s[10];
        for(int i=0;i<q;i++){
            scanf("%s",s);
            if(s[0]=='Q'){
                int l,r,k;
                scanf("%d%d%d",&l,&r,&k);
                p[i]=(que){0,l,r,k};
            }
            else {
                int l,r;
                scanf("%d%d",&l,&r);
                p[i]=(que){1,l,r,0};
                b[tot++]=r;
            }
        }
        sort(b,b+tot);
        tot=unique(b,b+tot)-b;
        root[0]=build(1,tot);
        for(int i=1;i<=n;i++){
            int pos=lower_bound(b,b+tot,a[i])-b+1;
            root[i]=update(root[i-1],1,tot,pos,1);
        }
        for(int i=1;i<=n;i++) bit[i]=root[0];
        for(int i=0;i<q;i++){
            if(p[i].op==0){
                for(int j=p[i].l-1;j;j-=lowbit(j)) use[j]=bit[j];
                for(int j=p[i].r;j;j-=lowbit(j)) use[j]=bit[j];
                printf("%d\n",b[query(p[i].l-1,p[i].r,root[p[i].l-1],root[p[i].r],1,tot,p[i].k)-1]);
            }
            else{
                int pos1=lower_bound(b,b+tot,a[p[i].l])-b+1;
                int pos2=lower_bound(b,b+tot,p[i].r)-b+1;
                modify(p[i].l,pos1,-1);
                modify(p[i].l,pos2,1);
                a[p[i].l]=p[i].r;
            }
        }
    return 0;
}
发布了323 篇原创文章 · 获赞 31 · 访问量 22万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览