BZOJ1901: Zju2112 Dynamic Rankings
题解:很显然的树套树 主席树+树状数组 动态维护区间权值 然后类似于查询区间第K大 复杂度nlog^2n 空间也如此
#include <bits/stdc++.h>
const int MAXN=2e4+10;
using namespace std;
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return f*x;
}
typedef struct node{
int l,r,sum;
}node;
node d[MAXN*201];
int rt[MAXN];
int cnt;
int get_id(int x){return x&(-x);}
void update(int &x,int y,int l,int r,int t,int vul){
x=++cnt;d[x]=d[y];d[x].sum+=vul;
if(l==r)return ;
int mid=(l+r)>>1;
if(t<=mid)update(d[x].l,d[y].l,l,mid,t,vul);
else update(d[x].r,d[y].r,mid+1,r,t,vul);
}
int ans,sum;
vector<int>v1;
vector<int>v2;
void querty(int l,int r,int k){
if(l==r){ans=l;return ;}
sum=0;
int mid=(l+r)>>1;
for(int i=0;i<v1.size();i++)sum+=d[d[v1[i]].l].sum;
for(int i=0;i<v2.size();i++)sum-=d[d[v2[i]].l].sum;
// cout<<l<<" "<<r<<" "<<sum<<endl;
if(k<=sum){
for(int i=0;i<v1.size();i++)v1[i]=d[v1[i]].l;
for(int i=0;i<v2.size();i++)v2[i]=d[v2[i]].l;
querty(l,mid,k);
}
else{
for(int i=0;i<v1.size();i++)v1[i]=d[v1[i]].r;
for(int i=0;i<v2.size();i++)v2[i]=d[v2[i]].r;
querty(mid+1,r,k-sum);
}
}
vector<int>vec;
int a[MAXN];
typedef struct Q{
int op,l,r,k;
}Q;
Q q[MAXN];
int main(){
int n,m;n=read();m=read();
for(int i=1;i<=n;i++)a[i]=read(),vec.push_back(a[i]);
char ch;
for(int i=1;i<=m;i++){
scanf(" %c",&ch);
if(ch=='Q'){q[i].op=1;q[i].l=read();q[i].r=read();q[i].k=read();}
else{q[i].op=2;q[i].l=read();q[i].r=read();vec.push_back(q[i].r);}
}
sort(vec.begin(),vec.end());
int sz=unique(vec.begin(),vec.end())-vec.begin();
for(int i=1;i<=n;i++){
a[i]=lower_bound(vec.begin(),vec.begin()+sz,a[i])-vec.begin()+1;
// cout<<a[i]<<" ";
for(int j=i;j<=n;j+=get_id(j))update(rt[j],rt[j],1,sz,a[i],1);
// cout<<endl;
}
// cout<<endl;
for(int i=1;i<=m;i++){
if(q[i].op==2){
q[i].r=lower_bound(vec.begin(),vec.begin()+sz,q[i].r)-vec.begin()+1;
for(int j=q[i].l;j<=n;j+=get_id(j))update(rt[j],rt[j],1,sz,a[q[i].l],-1),update(rt[j],rt[j],1,sz,q[i].r,1);
a[q[i].l]=q[i].r;
}
else{
v1.clear();v2.clear();
// cout<<q[i].l<<" "<<q[i].r<<" "<<q[i].k<<endl;
for(int j=q[i].r;j>0;j-=get_id(j))v1.push_back(rt[j]);
// cout<<endl;
for(int j=q[i].l-1;j>0;j-=get_id(j))v2.push_back(rt[j]);
// cout<<endl;
querty(1,sz,q[i].k);
printf("%d\n",vec[ans-1]);
}
}
return 0;
}
1901: Zju2112 Dynamic Rankings
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 9085 Solved: 3802
[Submit][Status][Discuss]
Description
给定一个含有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继续回答上面的问题。
Input
第一行有两个正整数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
m,n≤10000
Output
对于每一次询问，你都需要输出他的答案，每一个输出占单独的一行。
Sample Input
5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
Sample Output
3
6
6
