【bzoj1901】Zju2112 Dynamic Rankings 树状数组套主席树

Description

给定一个含有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。

Input

对于每一次询问，你都需要输出他的答案，每一个输出占单独的一行。

Output

Sample Input

5 3

3 2 1 4 7

Q 1 4 3

C 2 6

Q 2 5 3

Sample Output

3

6

HINT

20%的数据中，m,n≤100; 40%的数据中，m,n≤1000; 100%的数据中，m,n≤10000。

Source

裸题…

又被内存池大小坑了…

感觉这样写很不滋磁啊，有没有更好的写法？

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

typedef long long LL;
const int SZ = 1000010;
const int INF = 1000000010;
const double eps = 1e-6;

struct haha{
    char opt[3];
    int a,b,c;
}ask[SZ];

int num[SZ],bits[SZ],len,n,m;

int lsh[SZ];

struct node{
    int l,r;
    int cnt;
}tree[5000000];

int Tcnt = 0;

int rt[SZ];

void insert(int l,int r,int last,int &now,int v,int x)
{
    now = ++ Tcnt;
    tree[now] = tree[last];
    tree[now].cnt += x;
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(v <= mid) insert(l,mid,tree[last].l,tree[now].l,v,x);
    else insert(mid + 1,r,tree[last].r,tree[now].r,v,x);
}

void change(int pos,int v,int x)
{
    for(int i = pos;i <= n;i += (i & -i))
        insert(1,len,rt[i],rt[i],v,x);
}

int tl,tr;

int tml[SZ],tmr[SZ];

int query(int l,int r,int k)
{
    if(l == r) return l;
    int suml = 0,sumr = 0;
    for(int i = 1;i <= tl;i ++)
        suml += tree[tree[tml[i]].l].cnt;
    for(int i = 1;i <= tr;i ++)
        sumr += tree[tree[tmr[i]].l].cnt;
    int mid = l + r >> 1;
    if(sumr - suml >= k)
    {
        for(int i = 1;i <= tl;i ++)
            tml[i] = tree[tml[i]].l;
        for(int i = 1;i <= tr;i ++)
            tmr[i] = tree[tmr[i]].l;
        return query(l,mid,k);
    }
    else
    {
        for(int i = 1;i <= tl;i ++)
            tml[i] = tree[tml[i]].r;
        for(int i = 1;i <= tr;i ++)
            tmr[i] = tree[tmr[i]].r;
        return query(mid + 1,r,k - (sumr - suml));      
    }
}

int askk(int l,int r,int k)
{
    l --;
    tl = tr = 0;
    for(int i = l;i > 0;i -= (i & -i))
        tml[++ tl] = rt[i];
    for(int i = r;i > 0;i -= (i & -i))
        tmr[++ tr] = rt[i];
    return query(1,len,k);
}



int main()
{
//  freopen("1901.in","r",stdin);
//  freopen("1901.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i ++)
    {
        scanf("%d",&num[i]);
        lsh[++ lsh[0]] = num[i];
    }
    for(int i = 1;i <= m;i ++)
    {
        scanf("%s",ask[i].opt);
        if(ask[i].opt[0] == 'Q')
            scanf("%d%d%d",&ask[i].a,&ask[i].b,&ask[i].c);
        else
            scanf("%d%d",&ask[i].a,&ask[i].b),lsh[++ lsh[0]] = ask[i].b;
    }
    sort(lsh + 1,lsh + 1 + lsh[0]);
    len = unique(lsh + 1,lsh + 1 + lsh[0]) - lsh - 1;

    for(int i = 1;i <= n;i ++)
    {
        int x = lower_bound(lsh + 1,lsh + 1 + len,num[i]) - lsh;
        change(i,x,1);
    }

    for(int i = 1;i <= m;i ++)
    {
        if(ask[i].opt[0] == 'Q')
        {
            printf("%d\n",lsh[askk(ask[i].a,ask[i].b,ask[i].c)]);
        }
        else
        {
            int pos = ask[i].a;
            int x = lower_bound(lsh + 1,lsh + 1 + len,num[pos]) - lsh;
            change(pos,x,-1);
            num[pos] = ask[i].b;
            x = lower_bound(lsh + 1,lsh + 1 + len,num[pos]) - lsh;
            change(pos,x,1);
        }
    }

    return 0;
}