题目描述
给定一个含有n个数的序列a[1],a[2],a[3]……a[n],程序必须回答这样的询问:对于给定的i,j,k,在a[i],a[i+1],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1),并且,你可以改变一些a[i]的值,改变后,程序还能针对改变后的a继续回答上面的问题。你需要编一个这样的程序,从输入文件中读入序列a,然后读入一系列的指令,包括询问指令和修改指令。
对于每一个询问指令,你必须输出正确的回答。
输入输出格式
输入格式:第一行有两个正整数n(1≤n≤10000),m(1≤m≤10000)。分别表示序列的长度和指令的个数。
第二行有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。
对于每一次询问,你都需要输出他的答案,每一个输出占单独的一行。
输入输出样例
5 3 3 2 1 4 7 Q 1 4 3 C 2 6 Q 2 5 3
3 6
说明
20%的数据中,m,n≤100;
40%的数据中,m,n≤1000;
100%的数据中,m,n≤10000。
来源:bzoj1901
本题数据为洛谷自造数据,使用CYaRon耗时5分钟完成数据制作。
最近学了主席树和树状数组,但来不及单独写博客,干脆放到一起写在这道树状数组套主席树的题。
别听这个名字有点高大上,但实际上并不是很难理解,下面我们来分析一下:
要想做到在主席树上执行修改操作,如果暴力改,由于主席树是前缀和,就要把要改的这个位置的后面的主席树全部重建,极端情况下修改的复杂度是O(n),复杂度极高,而且空间也无法承受。同样是前缀和,这时我们想到了树状数组,树状数组通过lowbit操作可以做到查询某一个位置的前缀和,而且树状数组的空间需求只是O(n)的,那么在这个题上我们便可以利用一下这个性质,在每一个节点都建立一颗主席树,对于每一个位置的数,插入时只需要用lowbit操作把包含这个数的节点根据原先的节点信息维护一颗主席树,一样每一个点的只需新建log(n)个节点,总的新建的节点数就是log2(n)个节点,空间复杂度可以接受,至于查询,比如说查询l到r区间的第k大,就可以直接按主席树的方法查询,唯一的不同是要把lowbit到的树状数组节点保存,便于求前缀和,具体实现可以参考代码:
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define For(i,a,b) for(i=(a);i<=(b);++i)
#define rep(i,a,b) for(i=(a);i>=(b);--i)
#define ll long long
#define inf 999999999
#define mm(a,b) memset(a,b,sizeof(a))
using namespace std;
int read(){
int sum = 0,fg = 1;char c = getchar();
while(c < '0' || c > '9'){if(c =='-')fg = -1;c = getchar();}
while(c >='0' && c <='9')sum = sum*10 + c-'0',c = getchar();
return sum*fg;
}
int lowbit(int x){return x & -x;}
const int maxn = 6000010;
const int maxm = 10010;
int a[maxm] ,ls[maxm*2] ,cnt ,A[maxm] ,B[maxm] ,K[maxm] ,tree[maxm] ,node;
int sum[maxn] ,l[maxn] ,r[maxn] ,lcnt, rcnt, lsl[maxm], lsr[maxm] ,n ,m;
char ss[5] ;
void build(int s , int e , int x , int &y , int pos , int type){//普通的主席树insert
y = ++ node;
l[y] = l[x], r[y] = r[x];
sum[y] = sum[x] + type;
if(s == e)return ;
int mid = s + e >> 1;
if(pos <= mid){
build(s , mid , l[x] , l[y] , pos , type);
}
else build(mid + 1, e , r[x] , r[y] , pos , type);
}
int query(int s,int e,int k){
if(s == e)return s;
int mid = s+e >> 1;
int suml = 0, sumr = 0;
for(int i = 1;i <= lcnt; ++i){
suml += sum[l[lsl[i]]];//把1到l的左子树的和累加
}
for(int i = 1;i <= rcnt; ++i){
sumr += sum[l[lsr[i]]];
}
if(sumr - suml >= k){
for(int i = 1;i <= lcnt; ++i){
lsl[i] = l[lsl[i]];
}
for(int i = 1;i <= rcnt; ++i){
lsr[i] = l[lsr[i]];
}
return query(s , mid , k);
}
else {
for(int i = 1;i <= lcnt; ++i){
lsl[i] = r[lsl[i]];
}
for(int i = 1;i <= rcnt; ++i){
lsr[i] = r[lsr[i]];
}
return query(mid+1, e, k - sumr + suml);
}
}
int main(){
#ifndef ONLINE_JUDGE
freopen("THUNDER.in","r",stdin);
freopen("THUNDER.out","w",stdout);
#endif
int i,j;
n = read(),m = read();
For(i ,1 ,n){
a[i] = read();
ls[++cnt] = a[i];
}
For(i ,1 ,m){
scanf("%s", ss);
if(ss[0] == 'Q'){
A[i] = read(), B[i] = read(), K[i] = read();
}
else {
A[i] = read(), B[i] = read(); ls[++cnt] = B[i];//离线处理,因为要把中途要改变的权值也考虑进去,扩大主席树的宽度
}
}
sort(ls+1 , ls+cnt+1);
cnt = unique(ls+1, ls+cnt+1) - (ls+1);
For(i ,1 ,n){
int pos = lower_bound(ls+1 ,ls+cnt+1, a[i]) - ls;
for(j = i; j <= n; j += lowbit(j)){
build(1 , cnt , tree[j] , tree[j] , pos , 1);//用原来这个节点的信息,节省空间
}
}
For(i ,1 ,m){
if(K[i]){
lcnt = 0,rcnt = 0;
for(j = A[i] - 1; j ; j -= lowbit(j)){//为了求前缀和,要把遍历到的点加入数组
lsl[++lcnt] = tree[j];
}
for(j = B[i]; j ;j -= lowbit(j)){
lsr[++rcnt] = tree[j];
}
printf("%d\n", ls[query(1 , cnt ,K[i])]);
}
else{
int val = a[A[i]];
int pos = lower_bound(ls + 1,ls + cnt + 1,val) - ls;
for(j = A[i];j <= n; j += lowbit(j)){
build(1 , cnt, tree[j], tree[j] , pos, -1);//这里有点特殊,要先把原先这个数做更新的区间全部删掉信息,再在下面更新
}
a[A[i]] = B[i];
pos = lower_bound(ls + 1,ls + cnt + 1,B[i]) - ls;
for(j = A[i];j <= n; j += lowbit(j)){
build(1 , cnt, tree[j] , tree[j] , pos, 1);
}
}
}
return 0;
}