【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;
}

发布了307 篇原创文章 · 获赞 27 · 访问量 29万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览