[caioj1442][主席树][树状数组]第K大的数 II

【问题描述】

给n(1<=n<=50000)个数字，进行m(1<=m<=10000)次操作，有两种操作：
Q l r k：询问l到r第k小的数。
C x k：改变第x个数的值为k。

【输入文件】

第一行为n和m。 接下来一行n个数。 接下来m行为m个操作。

【输出文件】

遇到Q操作就输出。

【输入样例】

5 3
1 2 3 4 5
Q 2 5 2
C 4 9
Q 3 5 3

【输出样例】

3
9

【题解】

就和莫队一样。。修改要套一个东西。
why？因为你每个线段树用的是前一棵的东西。那么如果有一棵改了，后面所有都要暴力改掉。那么这样，我们就一起来TLE自动机吧！
仔细想想，我们主席树有一个c值，是用前缀和的思想求出来的。前缀和？好像有个叫树状数组的
对于n个值，我们建2n条链。前n条链维护的是没有修改的主席树，那么后n条就是维护修改的操作了。
对于修改x，我们找到x在后n条中对应的x+n，将他原来指向权值的那条链的值-1。那么当这条链与原链相加的时候，影响就消掉了。对应的，将点指向修改后的权值，并将那条链上的权全部+1.在相加的时候，就相当于改了啦
那么对于1~l的修改，求和，和l~r的计算，就用树状数组搞定

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{
    int lc,rc,c;
}tr[5110000];int len,n,m;
int lowbit(int x){return x&-x;}
void add(int &u,int l,int r,int p,int c)
{
    if(u==0)u=++len;
    tr[u].c+=c;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(p<=mid)add(tr[u].lc,l,mid,p,c);
    else add(tr[u].rc,mid+1,r,p,c);
}
void merge(int &x,int y)
{
    if(x==0){x=y;return ;}
    if(y==0)return ;
    tr[x].c+=tr[y].c;
    merge(tr[x].lc,tr[y].lc);
    merge(tr[x].rc,tr[y].rc);
}
int ust[210000],a[210000],rt[210000];
void turn(int x,int op)//-1 左孩子 1 右孩子 
{
    while(x>n)
    {
        if(op==0)ust[x]=rt[x];
        else if(op==-1)ust[x]=tr[ust[x]].lc;
        else ust[x]=tr[ust[x]].rc;
        x-=lowbit(x);
    }
}
void change(int x,int p,int c)
{
    while(x<=2*n)
    {
        add(rt[x],0,1000000000,p,c);
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    int ret=0;
    while(x>n)
    {
        ret+=tr[tr[ust[x]].lc].c;
        x-=lowbit(x);
    }
    return ret;
}
int findKth(int x,int y,int px,int py,int l,int r,int k)
{
    if(l==r)return l;
    int c=tr[tr[y].lc].c-tr[tr[x].lc].c+getsum(py+n);
    c-=getsum(px+n);
    int mid=(l+r)/2;
    if(k<=c)
    {
        turn(px+n,-1);turn(py+n,-1);
        return findKth(tr[x].lc,tr[y].lc,px,py,l,mid,k);
    }
    else
    {
        turn(px+n,1);turn(py+n,1);
        return findKth(tr[x].rc,tr[y].rc,px,py,mid+1,r,k-c);
    }
}
char ss[10];
int main()
{
    len=0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        add(rt[i],0,1000000000,a[i],1);
        merge(rt[i],rt[i-1]);
    }
    while(m--)
    {
        int x,y,k;
        scanf("%s",ss+1);
        if(ss[1]=='Q')
        {
            scanf("%d%d%d",&x,&y,&k);
            turn(x-1+n,0);turn(y+n,0);
            printf("%d\n",findKth(rt[x-1],rt[y],x-1,y,0,1000000000,k));
        }
        else
        {
            scanf("%d%d",&x,&y);
            change(x+n,a[x],-1);
            a[x]=y;
            change(x+n,a[x],1);
        }
    }
    return 0;
}