Bzoj 1901 Zju2112 Dynamic Rankings(树状数组+主席树)

转载请注明出处,谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

题目:给出一个序列,查询区间第K小,修改某个数

对于每一个位置建立主席树,和静态主席树不一样。
由于 有修改操作,每一棵主席树维护的只是某一个位置,而不是静态中的前缀和或者后缀和
而这里的前缀后就交给树状数组去维护。
也就是主席树维护所有位置的信息,树状数组维护前缀和。
代码还是非常直观的。
在ZOJ只给了32M内存,而且n<=50000, 交了几发没有过
在BZOJ上没有问题,感觉自己写的主席树比较耗内存
再研究下看看能不能改进
#include<iostream>  
#include<cstdio>  
#include<map>  
#include<cstring>  
#include<cmath>  
#include<vector>  
#include<algorithm>  
#include<set>  
#include<stack>
#include<string>  
#include<ctime>
#include<queue>  
#include<cassert>
#define inf 1000000005  
#define M 2560005 
#define N 61005
#define maxn 210005  
#define eps 1e-8
#define zero(a) fabs(a)<eps  
#define Min(a,b) ((a)<(b)?(a):(b))  
#define Max(a,b) ((a)>(b)?(a):(b))  
#define pb(a) push_back(a)  
#define mp(a,b) make_pair(a,b)  
#define mem(a,b) memset(a,b,sizeof(a))  
#define LL long long  
#define MOD 1000000007
#define sqr(a) ((a)*(a))  
#define Key_value ch[ch[root][1]][0]  
#define test puts("OK");  
#define pi acos(-1.0)
#define lowbit(x) ((-(x))&(x))
#define HASH1 1331
#define HASH2 10001
#define C   240  
#define vi vector<int>
#define TIME 10  
#pragma comment(linker, "/STACK:1024000000,1024000000")  
using namespace std;
int tot=0,n,m=0,q;
int a[N],h[N];
int T[N],c[M],lson[M],rson[M];
struct Question{
    int kind;
    int l,r,k;
    Question(){}
    Question(int k,int l,int r,int _k):kind(k),l(l),r(r),k(_k){}
}Q[N];
void Init_hash(int k){
    sort(h,h+k);
    m=unique(h,h+k)-h;
}
int hash(int x){
    return lower_bound(h,h+m,x)-h;
}
int bulid(int l,int r){
    int root=tot++;
    if(l!=r){
        int mid=(l+r)>>1;
        lson[root]=bulid(l,mid);
        rson[root]=bulid(mid+1,r);
    }
    return root;
}
int update(int root,int pos,int val){
    int newroot=tot++,tmp=newroot;
    int l=0,r=m-1;
    c[newroot]=c[root]+val;
    while(l<r){
        int mid=(l+r)>>1;
        if(pos<=mid){  
            lson[newroot]=tot++;rson[newroot]=rson[root];  
            newroot=lson[newroot];root=lson[root];  
            r=mid;  
        }  
        else{  
            rson[newroot]=tot++;lson[newroot]=lson[root];  
            newroot=rson[newroot];root=rson[root];  
            l=mid+1;  
        }  
        c[newroot]=c[root]+val;
    }
    return tmp;
}
int use[N];
void add(int x,int pos,int val){
    //更新的时候,维护树状数组
    for(int i=x;i<=n;i+=lowbit(i))
        T[i]=update(T[i],pos,val);
}
int sum(int x){
    int ret=0;
    for(int i=x;i;i-=lowbit(i))
        ret+=c[lson[use[i]]];
    return ret;
}
int query(int left,int right,int k){
    int l=0,r=m-1;
    for(int i=left-1;i;i-=lowbit(i)) use[i]=T[i];
    for(int i=right;i;i-=lowbit(i)) use[i]=T[i];
    while(l<r){
        int mid=(l+r)>>1;
        int tmp=sum(right)-sum(left-1);
        if(tmp>=k){
            r=mid;
            for(int i=left-1;i;i-=lowbit(i))
                use[i]=lson[use[i]];
            for(int i=right;i;i-=lowbit(i))
                use[i]=lson[use[i]];
        }
        else{
            l=mid+1;
            k-=tmp;
            for(int i=left-1;i;i-=lowbit(i))
                use[i]=rson[use[i]];
            for(int i=right;i;i-=lowbit(i))
                use[i]=rson[use[i]];
        }
    }
    return l;
}
void Init_tree(){
    T[0]=bulid(0,m-1);
    //所有位置初始为空树
    for(int i=1;i<=n;i++)
        T[i]=T[0];
    //更新所有位置
    for(int i=1;i<=n;i++)
        add(i,hash(a[i]),1);
}
int main(){
    int t=1;
    scanf("%d",&t);
    while(t--){
        tot=0;m=0;
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            h[m++]=a[i];
        }
        for(int i=0;i<q;i++){
            char str[5];int l,r,k;
            scanf("%s",str);
            if(str[0]=='Q'){
                scanf("%d%d%d",&l,&r,&k);
                Q[i]=Question(0,l,r,k);
            }
            else{
                scanf("%d%d",&l,&k);
                Q[i]=Question(1,l,0,k);
                h[m++]=k;
            }
        }
        Init_hash(m);
        Init_tree();
        for(int i=0;i<q;i++){
            if(Q[i].kind==0){
                printf("%d\n",h[query(Q[i].l,Q[i].r,Q[i].k)]);
            }
            else{
                add(Q[i].l,hash(a[Q[i].l]),-1);
                add(Q[i].l,hash(Q[i].k),1);
                a[Q[i].l]=Q[i].k;
            }
        }
    }
    return 0;
}


发布了430 篇原创文章 · 获赞 54 · 访问量 135万+
展开阅读全文

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

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

分享到微信朋友圈

×

扫一扫,手机浏览