bzoj 1901: Zju2112 Dynamic Rankings

新姿势，，这真是神题啊，用这道题研究的线段树套平衡树，然后又围观了树状数组套主席树，各种涨姿势2333

说一下树状数组套主席树吧。

对于树状数组的每个节点维护一个主席树，然后查询就是把普通主席树的查询改成在树状数组上，每一次查询lowbit，加起来就好了2333，修改也是一样的。

#include<bits/stdc++.h>
#define N 100005
#define M 10000005
#define LL long long
#define inf 0x3f3f3f3f
#define lowbit(x) x&(-x) 
using namespace std;               //(zoj2112)
inline int ra()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
int n,m,tot,top,sz;
int v[10001],num[20001],hash[20001];
int flag[10001],A[10001],B[10001],K[10001],root[10001];
int sum[2200001],ls[2200001],rs[2200001];
int L[30],R[30],a,b;
int find(int x)
{
    int l=1,r=tot,mid;
    while (l<=r)
    {
        int mid=l+r>>1;
        if (hash[mid]<x) l=mid+1;
        else r=mid-1;
    }
    return l;
}
void update(int last, int l, int r, int &rt, int w, int x)
{
    rt=++sz; 
    sum[rt]=sum[last]+x;
    if (l==r) return;
    ls[rt]=ls[last]; rs[rt]=rs[last];
    int mid=l+r>>1;
    if (w<=mid) update(ls[last],l,mid,ls[rt],w,x);
    else update(rs[last],mid+1,r,rs[rt],w,x);
}
int query(int l, int r, int k)
{
    if (l==r) return l;
    int suml=0,sumr=0;
    for (int i=1; i<=a; i++) suml+=sum[ls[L[i]]];
    for (int i=1; i<=b; i++) sumr+=sum[ls[R[i]]];
    int mid=l+r>>1;
    if (sumr-suml>=k)
    {
        for (int i=1; i<=a; i++) L[i]=ls[L[i]];
        for (int i=1; i<=b; i++) R[i]=ls[R[i]];
        return query(l,mid,k);
    }
    else
    {
        for (int i=1; i<=a; i++) L[i]=rs[L[i]];
        for (int i=1; i<=b; i++) R[i]=rs[R[i]];
        return query(mid+1,r,k-(sumr-suml));
    }
}
int main()
{
    n=ra(); m=ra();
    for (int i=1; i<=n; i++)
        v[i]=ra(),num[++top]=v[i];
    for (int i=1; i<=m; i++)
    {
        char s[3]; scanf("%s",s);
        A[i]=ra(); B[i]=ra();
        if (s[0]=='Q') K[i]=ra(),flag[i]=1;
            else num[++top]=B[i];
    }
    sort(num+1,num+top+1);
    hash[++tot]=num[1];
    for (int i=2; i<=top; i++)
        if (num[i]!=num[i-1])
            hash[++tot]=num[i];
    for (int i=1; i<=n; i++)
    {
        int t=find(v[i]);
        for (int j=i; j<=n; j+=lowbit(j))
            update(root[j],1,tot,root[j],t,1);
    }
    for (int i=1; i<=m; i++)
        if (flag[i])
        {
            a=0,b=0; A[i]--;
            for (int j=A[i]; j>0; j-=lowbit(j))
                L[++a]=root[j];
            for (int j=B[i]; j>0; j-=lowbit(j))
                R[++b]=root[j];
            printf("%d
",hash[query(1,tot,K[i])]);
        }
        else
        {
            int t=find(v[A[i]]);
            for (int j=A[i]; j<=n; j+=lowbit(j))
                update(root[j],1,tot,root[j],t,-1);
            v[A[i]]=B[i];
            t=find(B[i]);
            for (int j=A[i]; j<=n; j+=lowbit(j))
                update(root[j],1,tot,root[j],t,1);
        }
        return 0;
}