可修改主席树 算法总结

题目：dynamic rankings
折磨了我一天的树状数组套主席树，可算是搞懂一些；
所以马上来写一些东西防止我以后再忘掉；
带修主席树总体思路就是让树状数组的每一个节点都是一颗主席树，由于主席树和树状数组都是利用前缀和思想实现的，所以树状数组上的求和与多颗主席树的求和等价；
再加点优化，先建一颗静态主席树，然后在树状数组上维护修改量；
感觉这题代码上不加点注释很快我自己就忘了orz

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdio>
#define LL long long
#define random(a,b) (a+rand()%(b-a+1))
const int maxn=10005;
int n,m,num;
int too[maxn];
int s[maxn]; 
int a[maxn*2],b[maxn];
char c[3];
int root[maxn];
int lowbit(int x)
{
    return x &(-x);
}
struct asd
{
    int l,r,sum;
}t[maxn*1000];
struct qwe
{
    int l,r,tk,ck;
}q[maxn];
int ntot=0;
void build(int l,int r,int &x,int y,int pos,int v)
{
    t[++ntot]=t[y],x=ntot,t[x].sum+=v;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(pos<=mid)build(l,mid,t[x].l,t[y].l,pos,v);
    else build(mid+1,r,t[x].r,t[y].r,pos,v);
}
int lc(int x)
{
    return std::lower_bound(a+1,a+num+1,x)-a;
}
void insert(int timeline,int pos,int v)//timeline是在序列中的位置，pos是修改的数的排名，v是加数还是减数
{
    for(int i=timeline;i<=n;i+=lowbit(i))//外层的树状数组
        build(1,num,s[i],s[i],pos,v);
}
int getsum(int x)
{
    int ans=0;
    for(int i=x;i;i-=lowbit(i))
    ans+=t[t[too[i]].l].sum;//加的值是左儿子的值，便于二分，因此too指向的是当前节点
    return ans;
}
int  query(int l,int r,int nl,int nr,int nk,int gl,int gr)
{
    for(int i=gl;i;i-=lowbit(i))too[i]=s[i];
    for(int i=gr;i;i-=lowbit(i))too[i]=s[i];//注意这里是可以重叠的，因为所需记录的值都一样；
    while(l!=r)
    {
        int tsum=t[t[nr].l].sum-t[t[nl].l].sum-getsum(gl)+getsum(gr);//这里tsum的值是左儿子的和，便于二分
        int mid=(l+r)>>1;
        if(tsum>=nk)
        {
            for(int i=gl;i;i-=lowbit(i))too[i]=t[too[i]].l;
            for(int i=gr;i;i-=lowbit(i))too[i]=t[too[i]].l;//for循环仍是gl和gr（即最初询问的l-1和r）将too指向的移向其儿子
            r=mid;
            nl=t[nl].l;
            nr=t[nr].l;
        }
        else
        {
            for(int i=gl;i;i-=lowbit(i))too[i]=t[too[i]].r;
            for(int i=gr;i;i-=lowbit(i))too[i]=t[too[i]].r;
            l=mid+1;
            nk-=tsum;
            nl=t[nl].r;
            nr=t[nr].r;
        }
     } 
    return l;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    num=n;
    for(int i=1;i<=m;i++)
    {
        scanf("%s",c);
        if(c[0]=='Q')
            scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].tk);
        else 
        {
            scanf("%d%d",&q[i].l,&q[i].tk);
            q[i].ck=1;
            a[++num]=q[i].tk; 
        }
    }
    std::sort(a+1,a+num+1);
    num=std::unique(a+1,a+num+1)-a-1;
    for(int i=1;i<=n;i++)build(1,num,root[i],root[i-1],lc(b[i]),1);
    for(int i=1;i<=m;i++)
    {
        if(q[i].ck)
        {
            insert(q[i].l,lc(b[q[i].l]),-1);
            insert(q[i].l,lc(q[i].tk),1);
            b[q[i].l]=q[i].tk;//原来序列的值也要单点修改
        }
        else
            printf("%d\n",a[query(1,num,root[q[i].l-1],root[q[i].r],q[i].tk,q[i].l-1,q[i].r)]);
    }
    return 0;
}