[bzoj1901]动态区间k大

定一个含有n个数的序列a[1],a[2],a[3]……a[n]，程序必须回答这样的询问：对于给定的i,j,k，在a[i],a[i+1],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1)，并且，你可以改变一些a[i]的值，改变后，程序还能针对改变后的a继续回答上面的问题。你需要编一个这样的程序，从输入文件中读入序列a，然后读入一系列的指令，包括询问指令和修改指令。对于每一个询问指令，你必须输出正确的回答。 第一行有两个正整数n(1≤n≤10000)，m(1≤m≤10000)。分别表示序列的长度和指令的个数。第二行有n个数，表示a[1],a[2]……a[n]，这些数都小于10^9。接下来的m行描述每条指令，每行的格式是下面两种格式中的一种。 Q i j k 或者 C i t Q i j k （i,j,k是数字，1≤i≤j≤n, 1≤k≤j-i+1）表示询问指令，询问a[i]，a[i+1]……a[j]中第k小的数。C i t (1≤i≤n，0≤t≤10^9)表示把a[i]改变成为t。

写了一个树状数组套主席树的做法 nlog^2n, 还有一个节省空间但是耗时间的算法是线段树套平衡树，nlog^3n

沉迷数据结构

#include<iostream>
#include<cstdio>
#include<algorithm>
#define MN 2000000
#define MM 10000
using namespace std;
inline int read()
{
    int x =  , f = ; char ch = getchar();
    while(ch < '' || ch > ''){ if(ch == '-') f = -;  ch = getchar();}
    while(ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x * f;
}
char op[];
int rt[MM*+],n,m,cnt,tot,l2[MM*+],s[MM+],c[],cnum,b[],bnum;
struct Segment_Tree{
    int l,r,x;
}T[MN+];
struct ques{
    int kind,l,r,v;
}q[MM+];

void ins(int x,int v,int ad,int l=,int r=tot)
{
    T[x].x+=ad;
    if(l==r) return;
    int mid=l+r>>;
    if(v<=mid) ins(T[x].l?T[x].l:(T[x].l=++cnt),v,ad,l,mid);
    else ins(T[x].r?T[x].r:(T[x].r=++cnt),v,ad,mid+,r);
}

void renew(int x,int v,int ad)
{
    for(;x<=tot;x+=x&(-x))
        ins(rt[x],v,ad);
}

int main()
{
    tot=n=read();m=read();
    for(int i=;i<=n;i++) l2[i]=s[i]=read();
    for(int i=;i<=m;i++)
    {
        scanf("%s",op);q[i].l=read();
        q[i].r=read();
        if(op[]=='Q') q[i].kind=,q[i].v=read();
        else q[i].kind=,l2[++tot]=q[i].r;
    }
    sort(l2+,l2+tot+);
    for(int i=;i<=tot;i++) rt[i]=i;cnt=tot;
    for(int i=;i<=n;i++)
    {
        s[i]=lower_bound(l2+,l2+tot+,s[i])-l2;
        renew(i,s[i],);
    }
    for(int i=;i<=m;i++)
    {
        if(q[i].kind==)
        {
            q[i].r=lower_bound(l2+,l2+tot+,q[i].r)-l2;
            renew(q[i].l,s[q[i].l],-);
            renew(q[i].l,q[i].r,);
            s[q[i].l]=q[i].r;
        }
        else
        {
            int l=,r=tot,mid;cnum=bnum=;
            for(int j=q[i].r;j;j-=j&(-j))c[++cnum]=rt[j];
            for(int j=q[i].l-;j;j-=j&(-j)) b[++bnum]=rt[j];
            while(l<=r)
            {
                mid=l+r>>;int x=;
                for(int j=;j<=cnum;j++) x+=T[T[c[j]].l].x;
                for(int j=;j<=bnum;j++) x-=T[T[b[j]].l].x;
                if(x>=q[i].v)
                {
                    r=mid;
                    for(int j=;j<=cnum;j++) c[j]=T[c[j]].l;
                    for(int j=;j<=bnum;j++) b[j]=T[b[j]].l;
                }
                else
                {
                    l=mid+;q[i].v-=x;
                    for(int j=;j<=cnum;j++) c[j]=T[c[j]].r;
                    for(int j=;j<=bnum;j++) b[j]=T[b[j]].r;
                }
            }
            printf("%d\n",l2[r]);
        }
    }
    return ;
}

线段树套平衡树  结果我一不小心写成了树状数组  空间nlogn 时间nlog^3n

一开始把重建操作注释掉了交上去居然也过了，BST最棒啦，垃圾替罪羊

#include<iostream>
#include<cstdio>
#include<algorithm>
#define MN 400000
#define MM 10000
using namespace std;
inline int read()
{
    int x =  , f = ; char ch = getchar();
    while(ch < '' || ch > ''){ if(ch == '-') f = -;  ch = getchar();}
    while(ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x * f;
}
char op[];
int rt[MM*+],n,m,cnt,mark=,tot,top=,l2[MM*+],s[MM+];
int c[MN+][],size[MN+],num[MN+],nn[MN+],fa[MN+],qx[MN+];
struct ques{
    int kind,l,r,v;
}q[MM+];

void ins(int&x,int v,int ad,int last)
{
    //cout<<"ins"<<x<<" "<<v<<" "<<ad<<" "<<last<<endl;
    if(!x){x=++cnt;nn[x]=v;num[x]=;size[x]=;fa[x]=last;return;}
    if(v==nn[x]){num[x]+=ad;size[x]+=ad;return;}
    if(v<nn[x]) ins(c[x][],v,ad,x);
    else ins(c[x][],v,ad,x);
    size[x]=size[c[x][]]+size[c[x][]]+num[x];
    if(max(size[c[x][]],size[c[x][]])>=0.7*size[x]) mark=x;
}

void dfs(int x)
{
    if(c[x][]) dfs(c[x][]);
    if(num[x])qx[++top]=x;
    if(c[x][]) dfs(c[x][]);
    fa[x]=c[x][]=c[x][]=;
}

void build(int&x,int l,int r,int last)
{
    if(l>r) {x=;return;}int mid=l+r>>;
    x=qx[mid];fa[x]=last;
    build(c[x][],l,mid-,x);
    build(c[x][],mid+,r,x);
    size[x]=size[c[x][]]+size[c[x][]]+num[x];
}

void rebuild(int i)
{
    top=;int y=fa[mark];dfs(mark);
    if(!y) build(rt[i],,top,);
    else build(c[y][c[y][]==mark],,top,y);
    mark=;
}

void renew(int x,int v,int ad)
{
    for(;x<=tot;x+=x&(-x))
    {
        ins(rt[x],v,ad,);
        if(mark) rebuild(x);
    }
}

int query(int x,int k)
{
    if(!x) return ;
    if(nn[x]>=k) return query(c[x][],k)+((nn[x]==k)?num[x]:);
    else return size[c[x][]]+num[x]+query(c[x][],k);
} 

int check(int x,int v)
{
    int sum=;
    for(;x;x-=x&(-x))
        sum+=query(rt[x],v);
    return sum;
}

int main()
{
    tot=n=read();m=read();
    for(int i=;i<=n;i++) l2[i]=s[i]=read();
    for(int i=;i<=m;i++)
    {
        scanf("%s",op);q[i].l=read();
        q[i].r=read();
        if(op[]=='Q') q[i].kind=,q[i].v=read();
        else q[i].kind=,l2[++tot]=q[i].r;
    }
    sort(l2+,l2+tot+);
    for(int i=;i<=n;i++)
    {
        s[i]=lower_bound(l2+,l2+tot+,s[i])-l2;
        renew(i,s[i],);
    }
    for(int i=;i<=m;i++)
    {
        if(q[i].kind==)
        {
            q[i].r=lower_bound(l2+,l2+tot+,q[i].r)-l2;
            renew(q[i].l,s[q[i].l],-);
            renew(q[i].l,q[i].r,);
            s[q[i].l]=q[i].r;
        }
        else
        {
            int l=,r=tot,mid,ans=;
            while(l<=r)
            {
                mid=l+r>>;
                int x=check(q[i].r,mid)-check(q[i].l-,mid);
                //cout<<l2[mid]<<" "<<check(q[i].r,mid)<<" "<<check(q[i].l-1,mid)<<endl;
                if(x>=q[i].v) ans=mid,r=mid-;
                else l=mid+;
            }
            printf("%d\n",l2[ans]);
        }
    }
    return ;
}