码迷,mamicode.com
首页 > 其他好文 > 详细

动态开点的权值线段树搞普通平衡树。。。补发一波

时间:2019-07-03 20:19:55      阅读:129      评论:0      收藏:0      [点我收藏+]

标签:ifd   sdi   struct   stream   nbsp   efi   def   stdin   sig   

RT.

#include<iostream>
#include<cstdio>
#define R register int
using namespace std; 
inline int g() {
    R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch==-?-1:fix;
    do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix;
}
const int N=100010,Inf=1E+7+10; 
struct node {int ls,rs,cnt;}t[N<<2];
int n,tot,rt;
#define ls t[tr].ls
#define rs t[tr].rs
#define cnt(x) t[x].cnt
inline void add(int& tr,int l,int r,int vl,int d) {
    if(!tr) tr=++tot; if(l==r) {cnt(tr)+=d; return ;} R md=l+r>>1;
    if(vl<=md) add(ls,l,md,vl,d); else add(rs,md+1,r,vl,d); cnt(tr)=cnt(ls)+cnt(rs);
}
inline int query(int tr,int l,int r,int LL,int RR) {
    if(!tr) return 0; if(LL<=l&&r<=RR) return cnt(tr); R md=l+r>>1,ret=0;
    if(LL<=md) ret+=query(ls,l,md,LL,RR); if(RR>md) ret+=query(rs,md+1,r,LL,RR); return ret;
}
inline int getvl(int tr,int l,int r,int rk) {
    if(l==r) return l; R md=l+r>>1;
    if(cnt(ls)>=rk) return getvl(ls,l,md,rk); else return getvl(rs,md+1,r,rk-cnt(ls));
}
inline int getrk(int tr,int l,int r,int vl) { if(!tr) return 0;
    if(l==r) return 1; R md=l+r>>1;
    if(vl<=md) return getrk(ls,l,md,vl); else return cnt(ls)+getrk(rs,md+1,r,vl);
}
inline int getpre(int vl) {R rk=query(1,-Inf,Inf,-Inf,vl-1); return getvl(1,-Inf,Inf,rk);}
inline int getnxt(int vl) {R rk=query(1,-Inf,Inf,-Inf,vl)+1; return getvl(1,-Inf,Inf,rk);}
signed main() {
#ifdef JACK
    freopen("NOIPAK++.in","r",stdin);
#endif
    n=g(); for(R i=1,op,x;i<=n;++i) {
        op=g(),x=g(); if(op==1) add(rt,-Inf,Inf,x,1); else if(op==2) add(rt,-Inf,Inf,x,-1);
        else if(op==3) printf("%d\n",getrk(1,-Inf,Inf,x)); 
        else if(op==4) printf("%d\n",getvl(1,-Inf,Inf,x));
        else if(op==5) printf("%d\n",getpre(x));
        else if(op==6) printf("%d\n",getnxt(x));    
    }
}

2019.07.03

 

动态开点的权值线段树搞普通平衡树。。。补发一波

标签:ifd   sdi   struct   stream   nbsp   efi   def   stdin   sig   

原文地址:https://www.cnblogs.com/Jackpei/p/11128437.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!