带修改主席树 洛谷P2617 Dynamic Rankings
题目大意:这道题依然是求区间第k大,但是这是求动态区间第k大的值,数组里的值可以随意修改。
输入:
第一行有两个正整数n(1≤n≤100000),m(1≤m≤100000)。分别表示序列的长度和指令的个数。
第二行有n个数,表示a[1],a[2]……a[n],这些数都小于10^9。接下来的m行描述每条指令,每行的格式是下面两种格式中的一种。 Q i j k 或者 C i t
- Q i j k (i,j,k是数字,1≤i≤j≤n, 1≤k≤j-i+1)表示询问指令,询问a[i],a[i+1]……a[j]中第k小的数。
- C i t (1≤i≤n,0≤t≤10^9)表示把a[i]改变成为t。
这是才学主席树没多久,就提枪上阵然后拜下阵来,这是主席树的变式,带修改的主席树,在主席树上面套一个树状数组,利用树状数组求得每一个状态下主席树的值。因为加上了一个树状数组,所以时间复杂度为n (logn)^2
嘛,我觉得代码上的注释讲得很清楚了,有不懂的可以在评论区里留言哈
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+7,LOG=600;
char c[maxn];
int n,m,tot=0,ztot=0;
int a[maxn],b[maxn],l[maxn],r[maxn],k[maxn];
int tree[maxn],Size[maxn*LOG],L[maxn*LOG],R[maxn*LOG];
int sx,sy,xx[maxn],yy[maxn];
void add(int &tr,int l,int r,int x,int num){
if(!tr) tr=++ztot;
Size[tr]+=num;
if(l==r)
return;
int mid=(l+r)>>1;
if(x<=mid) add(L[tr],l,mid,x,num);
else add(R[tr],mid+1,r,x,num);
}//将元素加入,对应的节点更新
void update(int x,int y){
int k=lower_bound(b+1,b+1+tot,a[x])-b;//离散化后的值
while(x<=n){
add(tree[x],1,tot,k,y);
x+=(x&-x);
}//同样更新树状数组中每一个状态的值
}
int query(int l,int r,int k){
if(l==r)
return l;
int sum=0,mid=(l+r)>>1;
for(int i=1;i<=sx;i++) sum-=Size[L[xx[i]]];
for(int i=1;i<=sy;i++) sum+=Size[L[yy[i]]];//利用树状数组算得每一个状态下的左子树之和即为真实的当前状态下的左子树的Size值
if(k<=sum){
for(int i=1;i<=sx;i++) xx[i]=L[xx[i]];
for(int i=1;i<=sy;i++) yy[i]=L[yy[i]];//进入左子树,当然要将左子树的状态更新过来
return query(l,mid,k);
}else{
for(int i=1;i<=sx;i++) xx[i]=R[xx[i]];
for(int i=1;i<=sy;i++) yy[i]=R[yy[i]];//同理,将右子树的状态更新过来
return query(mid+1,r,k-sum);
}
}
int main(){
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]),b[++tot]=a[i];
for(int i=1;i<=m;i++){
scanf(" %c %d %d",&c[i],&l[i],&r[i]);
if(c[i]=='Q')
scanf("%d",&k[i]);
else if(c[i]='C')
b[++tot]=r[i];//将需要改动的数也一起加入进行离散化
}
sort(b+1,b+1+tot);
tot=unique(b+1,b+1+tot)-b-1;
for(int i=1;i<=n;i++)
update(i,1);//首先将原来的数加入主席树(权值线段树)
for(int i=1;i<=m;i++){
if(c[i]=='C'){
update(l[i],-1);//需要修改的时候,将原来的数删去
a[l[i]]=r[i];//将原数组更改
update(l[i],1);//将改过的数加进去
}
else{
sx=0,sy=0;
for(int j=l[i]-1;j;j-=(j&-j)) xx[++sx]=tree[j];
for(int j=r[i];j;j-=(j&-j)) yy[++sy]=tree[j];//利用树状数组求得每一个状态下的主席树根节点
printf("%d\n",b[query(1,tot,k[i])]);//询问查找位置
}
}
return 0 ;
}