mcfx's blog - 树套树
/category/sts/
-
BZOJ 3196: Tyvj 1730 二逼平衡树
/archives/80/
2016-11-17T00:28:00+08:00
###Description
您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:
1.查询k在区间内的排名
2.查询区间内排名为k的值
3.修改某一位值上的数值
4.查询k在区间内的前驱(前驱定义为小于x,且最大的数)
5.查询k在区间内的后继(后继定义为大于x,且最小的数)
###Input
第一行两个数 n,m 表示长度为n的有序序列和m个操作
第二行有n个数,表示有序序列
下面有m行,opt表示操作标号
若opt=1 则为操作1,之后有三个数l,r,k 表示查询k在区间[l,r]的排名
若opt=2 则为操作2,之后有三个数l,r,k 表示查询区间[l,r]内排名为k的数
若opt=3 则为操作3,之后有两个数pos,k 表示将pos位置的数修改为k
若opt=4 则为操作4,之后有三个数l,r,k 表示查询区间[l,r]内k的前驱
若opt=5 则为操作5,之后有三个数l,r,k 表示查询区间[l,r]内k的后继
###Output
对于操作1,2,4,5各输出一行,表示查询结果
###Sample Input
9 6
4 2 2 1 9 4 0 1 1
2 1 4 3
3 4 10
2 1 4 3
1 2 5 9
4 3 9 5
5 2 8 5
###Sample Output
2
4
3
4
9
###HINT
1.n和m的数据范围:n,mb?a:b;}
template inline T min(T a,T b){return a0?a:-a;}
template inline void repr(T &a,T b){if(ab)a=b;}
template T gcd(T a,T b){if(b)return gcd(b,a%b);return a;}
#define mp(a,b) std::make_pair(a,b)
#define pb push_back
#define lb(x) ((x)&(-(x)))
#define sqr(x) ((x)*(x))
struct node
{
node *lc,*rc;
int cnt;
node();
}_null,*null=&_null;
node::node(){lc=rc=null;cnt=0;}
int pf;
void erase(node *x)
{
if(x->lc!=null)erase(x->lc);
if(x->rc!=null)erase(x->rc);
delete x;
}
void modify(node *&x,int l,int r,int p,int v)
{
if(x==null)x=new node;
x->cnt+=v;
if(!x->cnt)
{
erase(x);
x=null;
return;
}
if(l!=r)
{
int f=(l+r)/2;
if(plc,l,f,p,v);
else
modify(x->rc,f+1,r,p,v);
}
}
struct group
{
node *x[50];
int mul[50],sz;
inline int cnt()
{
int ret=0;
for(int i=0;icnt*mul[i];
return ret;
}
inline group* lc(group *f)
{
f->sz=sz;
for(int i=0;ix[i]=x[i]->lc,f->mul[i]=mul[i];
return f;
}
inline group* rc(group *f)
{
f->sz=sz;
for(int i=0;ix[i]=x[i]->rc,f->mul[i]=mul[i];
return f;
}
};
int cnt(group *x,int l,int r,int ql,int qr)
{
if(l==ql&&r==qr)return x->cnt();
int t=(l+r)/2,ans=0;
group ch;
if(qllc(&ch),l,t,ql,min(t,qr));
if(qr>t)ans+=cnt(x->rc(&ch),t+1,r,max(ql,t+1),qr);
return ans;
}
int kth(group *x,int l,int r,int rk)
{
if(l==r)return l;
group ch;
x->lc(&ch);
int lcnt;
if((lcnt=ch.cnt())>=rk)
return kth(&ch,l,(l+r)/2,rk);
else
return kth(x->rc(&ch),(l+r)/2+1,r,rk-lcnt);
}
int gmax(group *x,int l,int r,int p)
{
if(!x->cnt())return 0;
if(l==r)return l;
int t=(l+r)/2;
group ch;
if(plc(&ch),l,t,p);
if(r==p)
{
x->rc(&ch);
if(ch.cnt())return gmax(&ch,t+1,r,p);
return gmax(x->lc(&ch),l,t,t);
}
return max(gmax(x->lc(&ch),l,t,t),gmax(x->rc(&ch),t+1,r,p));
}
int gmin(group *x,int l,int r,int p)
{
if(!x->cnt())return 0x7fffffff;
if(l==r)return l;
int t=(l+r)/2;
group ch;
if(p>t)return gmin(x->rc(&ch),t+1,r,p);
if(l==p)
{
x->lc(&ch);
if(ch.cnt())return gmin(&ch,l,t,p);
return gmin(x->rc(&ch),t+1,r,t+1);
}
return min(gmin(x->lc(&ch),l,t,p),gmin(x->rc(&ch),t+1,r,t+1));
}
#define nl 0
#define nr 100000001
node *root[50001];
int v[50001];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i