区间第k大 可修改主席树,树状数组套线段树。

#include <cstdio>
#include <vector>
using namespace std;

const int N = 100010;

struct Node {
    int cnt;
    Node *ls, *rs;
}pool[N * 200], *tail = pool, *roots[N], *null;

int n, q;
int aa[N];
Node *va[10000], *vb[10000];
int ca, cb;

void init() {
    null = ++tail;
    null->ls = null;
    null->rs = null;
    null->cnt = 0;
}
Node *newnode() {
    Node *nd = ++tail;
    nd->ls = null;
    nd->rs = null;
    nd->cnt = 0;
    return nd;
}
void modify( int lf, int rg, int pos, int delta ) {
    for( int t = 0; t < ca; t++ )
        va[t]->cnt += delta;
    if( lf == rg ) 
        return;
    int mid = (lf + rg)>>1;
    if( pos <= mid ) {
        for( int t = 0; t < ca; t++ ) {
            if( va[t]->ls == null ) 
                va[t]->ls = newnode();
            va[t] = va[t]->ls;
        }
        modify( lf, mid, pos, delta );
    } else {
        for( int t = 0; t < ca; t++ ) {
            if( va[t]->rs == null )
                va[t]->rs = newnode();
            va[t] = va[t]->rs;
        }
        modify( mid+1, rg, pos, delta );
    }
}

int query_seg( int lf, int rg, int k ) {
    if( lf == rg ) return lf;
    int mid = (lf + rg) >> 1;
    int lz = 0;
    for( int t = 0; t < ca; t++ ) lz += va[t]->ls->cnt;
    for( int t = 0; t < cb; t++ ) lz -= vb[t]->ls->cnt;
    if( k <= lz ) {
        for( int t = 0; t < ca; t++ ) va[t] = va[t]->ls;
        for( int t = 0; t < cb; t++ ) vb[t] = vb[t]->ls;
        return query_seg( lf, mid, k );
    } else {
        for( int t = 0; t < ca; t++ ) va[t] = va[t]->rs;
        for( int t = 0; t < cb; t++ ) vb[t] = vb[t]->rs;
        k -= lz;
        return query_seg( mid+1, rg, k );
    }
}
void modify( int u, int x, int delta ) {
    ca = 0;
    for( int i = u; i <= n; i += i & -i ) 
        va[ca++] = roots[i];
    modify( 1, n, x, delta );
}
void modify( int u, int x ) {
    modify( u, aa[u], -1 );
    modify( u, x, +1 );
    aa[u] = x;
}
int query( int l, int r, int k ) {
    ca = cb = 0;
    for( int i = r; i; i -= i & -i )
        va[ca++] = roots[i];
    for( int i = l - 1; i; i -= i & -i )
        vb[cb++] = roots[i];
    return query_seg(1,n,k);
}

int main() {
    scanf( "%d%d", &n, &q );
    for( int i = 1; i <= n; i++ )
        scanf( "%d", aa + i );
    init();
    for( int i = 1; i <= n; i++ )
        roots[i] = newnode();
    for( int i = 1; i <= n; i++ )
        modify( i, aa[i], +1 );
    while( q-- ) {
        char ss[100];
        scanf( "%s", ss );
        if( ss[0] == 'q' ) {
            int l, r, k;
            scanf( "%d%d%d", &l, &r, &k );
            printf( "%d\n", query(l,r,k) );
        } else {
            int pos, val;
            scanf( "%d%d", &pos, &val );
            modify( pos, val );
        }
    }
//  fprintf( stderr, "mem used %d%%\n", int(tail - pool) * 100 / int(sizeof(pool)/sizeof(pool[0])) );
}

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值