BZOJ 1901 树状数组+函数式线段树

题解：

带单点更新的区间第k大~

函数式线段树秒杀了~

与不带修改的函数式线段树唯一不同的在于线段树中的每个结点维护的都是这个位置的结点的树状数组值~

然后自己随便怎么胡搞啊神马的就nlog^2n了~

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>

#define N 110040
#define lowbit(x) x&-x

using namespace std;

int h[N],ls[N<<5],rs[N<<5],sum[N<<5];
int cnt,n,m,gs,num;
int a[N],bh[N];
int dat[N][4],ln[N],rn[N],lc,rc;

inline int newnode(int s,int l,int r)
{
    ++cnt; sum[cnt]=s; ls[cnt]=l; rs[cnt]=r;
    return cnt;
}

inline void build(int l,int r,int &rt)
{
    rt=newnode(0,0,0);
    if(l==r) return;
    int mid=(l+r)>>1;
    build(l,mid,ls[rt]);
    build(mid+1,r,rs[rt]);
}

inline void updata(int last,int pos,int val,int l,int r,int &rt)
{
    rt=newnode(sum[last]+val,ls[last],rs[last]);
    if(l==r) return;
    int mid=(l+r)>>1;
    if(pos<=mid) updata(ls[last],pos,val,l,mid,ls[rt]);
    else updata(rs[last],pos,val,mid+1,r,rs[rt]);
}

inline void modify(int x,int pos,int val)
{
    int tmp;
    while(x<=n)
    {
        updata(h[x],pos,val,1,num,tmp);
        h[x]=tmp;
        x+=lowbit(x);
    }
}

inline int query(int l,int r,int k)
{
    if(l==r) return l;
    int tr=0,tl=0;
    for(int i=1;i<=rc;i++) tr+=sum[ls[rn[i]]];
    for(int i=1;i<=lc;i++) tl+=sum[ls[ln[i]]];
    tr-=tl;
    int mid=(l+r)>>1;
    if(k<=tr)
    {
        for(int i=1;i<=rc;i++) rn[i]=ls[rn[i]];
        for(int i=1;i<=lc;i++) ln[i]=ls[ln[i]];
        return query(l,mid,k);
    }
    else
    {
        for(int i=1;i<=rc;i++) rn[i]=rs[rn[i]];
        for(int i=1;i<=lc;i++) ln[i]=rs[ln[i]];
        return query(mid+1,r,k-tr);
    }
}

inline int getans(int l,int r,int k)
{
    rc=lc=0;
    while(r)
    {
        rn[++rc]=h[r];
        r-=lowbit(r);
    }
    while(l)
    {
        ln[++lc]=h[l];
        l-=lowbit(l);
    }
    return query(1,num,k);
}

inline void read()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        bh[++gs]=a[i];
    }
    char str[4];
    for(int i=1;i<=m;i++)
    {
        scanf("%s",str);
        if(str[0]=='Q')
        {
            dat[i][0]=0;
            for(int j=1;j<=3;j++) scanf("%d",&dat[i][j]);
        }
        else
        {
            dat[i][0]=1;
            scanf("%d%d",&dat[i][1],&dat[i][2]);
            bh[++gs]=dat[i][2];
        }
    }
}

inline void go()
{
    sort(bh+1,bh+1+gs);
    num=unique(bh+1,bh+1+gs)-bh-1;
    for(int i=1;i<=n;i++) a[i]=lower_bound(bh+1,bh+1+num,a[i])-bh;
    build(1,num,h[0]);
    for(int i=1;i<=n;i++) modify(i,a[i],1);
    for(int i=1;i<=m;i++)
    {
        if(dat[i][0]==0) printf("%d\n",bh[getans(dat[i][1]-1,dat[i][2],dat[i][3])]);
        else
        {
            int pos=lower_bound(bh+1,bh+1+num,dat[i][2])-bh;
            modify(dat[i][1],a[dat[i][1]],-1);
            a[dat[i][1]]=pos;
            modify(dat[i][1],a[dat[i][1]],1);
        }
    }
}

int main()
{
    read(),go();
    return 0;
}