【树状数组套权值线段树】bzoj1901 Zju2112 Dynamic Rankings

谁再管这玩意叫树状数组套主席树我跟谁急

明明就是树状数组的每个结点维护一棵动态开结点的权值线段树而已

好吧，其实只有一个指针，指向该结点的权值线段树的当前结点

每次查询之前，要让指针指向根结点

不同结点的权值线段树之间毫无关联

可以看这个：http://blog.csdn.net/popoqqq/article/details/40108669?utm_source=tuicool

#include<cstdio>
#include<algorithm>
using namespace std;
struct Data
{
	int p,v;
}t[20010];
int e,en=1;
bool cmp(const Data &a,const Data &b)
{
	return a.v<b.v;
}
int n,m,a[20010],ma[20010],x[10010],y[10010],z[10010];
char op[10010][2];
struct Node{int v,ch[2];}T[20010*195];
int root[10010],now[2][10010];
void Init_root(int ql,int qr)
{
	--ql;
	for(;ql;ql-=(ql&(-ql))) now[0][ql]=root[ql];
	for(;qr;qr-=(qr&(-qr))) now[1][qr]=root[qr];
}
int qBIT(int K,int ql,int qr)
{
	int res=0;
	for(int x=ql-1;x;x-=(x&(-x))) res-=T[T[now[0][x]].ch[0]].v;
	for(int x=qr;x;x-=(x&(-x))) res+=T[T[now[1][x]].ch[0]].v;
	bool f=(res<K);
	for(int x=ql-1;x;x-=(x&(-x))) now[0][x]=T[now[0][x]].ch[f];
	for(;qr;qr-=(qr&(-qr))) now[1][qr]=T[now[1][qr]].ch[f];
	return res;
}
int Kth(int K,int ql,int qr,int l,int r)//K小值
{
    if(l==r) return l;
    int m=(l+r>>1),tmp;
    if((tmp=qBIT(K,ql,qr))>=K) return Kth(K,ql,qr,l,m);
    return Kth(K-tmp,ql,qr,m+1,r);
}
void Update(int p,int v,int cur,int l,int r)
{
	if(l==r)
	  {
	  	T[cur].v+=v;
	  	return;
	  }
	int m=(l+r>>1);
	if(p<=m)
	  {
	  	if(!T[cur].ch[0]) T[cur].ch[0]=++e;
	  	Update(p,v,T[cur].ch[0],l,m);
	  }
	else
	  {
	  	if(!T[cur].ch[1]) T[cur].ch[1]=++e;
	  	Update(p,v,T[cur].ch[1],m+1,r);
	  }
	T[cur].v=T[T[cur].ch[0]].v+T[T[cur].ch[1]].v;
}
void Update(int pp,int p,int v)
{
	for(;pp<=n;pp+=(pp&(-pp)))
	  Update(p,v,root[pp],1,en);
}
int main()
{
//	freopen("bzoj1901.in","r",stdin);
//	freopen("bzoj1901.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i)
	  scanf("%d",&t[i].v);
	e=n;
	for(int i=1;i<=m;++i)
	  {
	  	scanf("%s%d",op[i],&x[i]);
	  	if(op[i][0]==‘C‘)
	  	  scanf("%d",&t[++e].v);
	  	else
	  	  scanf("%d%d",&y[i],&z[i]);
	  }
	for(int i=1;i<=e;++i)
	  t[i].p=i;
	sort(t+1,t+e+1,cmp);
	ma[a[t[1].p]=1]=t[1].v;
	for(int i=2;i<=e;++i)
	  {
	  	if(t[i].v!=t[i-1].v)
	  	  ++en;
	  	ma[a[t[i].p]=en]=t[i].v;
	  }
	e=n;
	for(int i=1;i<=m;++i)
	  if(op[i][0]==‘C‘)
	    z[i]=a[++e];
	e=n;
	for(int i=1;i<=n;++i)
	  root[i]=i;
	for(int i=1;i<=n;++i)
	  Update(i,a[i],1);
	for(int i=1;i<=m;++i)
	  if(op[i][0]==‘C‘)
	    {
	      Update(x[i],a[x[i]],-1);
	      a[x[i]]=z[i];
	      Update(x[i],z[i],1);
	    }
	  else
	    {
	      Init_root(x[i],y[i]);
	      printf("%d\n",ma[Kth(z[i],x[i],y[i],1,en)]);
	    }
	return 0;
}