[POJ2761] Feed the dogs (Treap)

题目链接：http://poj.org/problem?id=2761

题目大意：给你n个数，m次查询，m次查询分别是a,b,k，查询下表从a到b的第k小元素是哪个。这m个区间不会互相包含。

Treap，自己学着写了个板子，留贴备用。

离线操作，将区间移动之，删除旧的添加新的。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <string>
#include <bitset>
#include <cmath>
#include <numeric>
#include <iterator>
#include <iostream>
#include <cstdlib>
#include <functional>
#include <queue>
#include <stack>
#include <list>
#include <ctime>
using namespace std;

const int inf = ~0U >> ;

typedef long long LL;

struct TreapNode {
    int key, val, siz;
    TreapNode* ch[];
    TreapNode() :val(), siz(){ key = rand() - ; ch[] = ch[] = NULL; }
    void rz(){
        siz = ;
        if (ch[]) siz += ch[]->siz;
        if (ch[]) siz += ch[]->siz;
    }
    ~TreapNode(){
        if (ch[]) delete ch[];
        if (ch[]) delete ch[];
    }
};

void rot(TreapNode*& rt, bool d){
    TreapNode* c = rt->ch[d];
    rt->ch[d] = c->ch[!d];
    c->ch[!d] = rt;
    rt->rz(); c->rz();
    rt = c;
}

void Insert(TreapNode*& rt, int val) {
    if (!rt){ rt = new TreapNode(); rt->val = val; return; }
    //if (val == rt->val) return;
    bool d = val > rt->val;
    Insert(rt->ch[d], val);
    if (rt->ch[d]->key < rt->key) rot(rt, d);
    else rt->rz();
}

void Delete(TreapNode*& rt, int val) {
    if (rt == NULL) return;
    if (rt->val == val) {
        if (rt->ch[] == NULL && rt->ch[] == NULL) {
            delete rt;
            rt = NULL;
            return;
        }
        else if (rt->ch[] == NULL) {
            rot(rt, );
            Delete(rt->ch[], val);
        }
        else if (rt->ch[] == NULL) {
            rot(rt, );
            Delete(rt->ch[], val);
        }
        else {
            bool d = rt->ch[]->key > rt->ch[]->key;
            rot(rt, d);
            Delete(rt->ch[!d], val);
        }
    }
    else {
        bool d = val > rt->val;
        Delete(rt->ch[d], val);
    }
    rt->rz();
}

int Select(TreapNode* rt, int k) {
    if (!rt) return -inf;
    if (rt->siz < k) return -inf;
    int r = ;
    if (!rt->ch[]) r = ;
    else r = rt->ch[]->siz;
    if (r == k - ) return rt->val;
    if (k -  < r) return Select(rt->ch[], k);
    return Select(rt->ch[], k - r - );
}

int n, m;
int a[], ans[];
//int ptr = 0;

struct bwl{
    int l, r, k, idx;
    bool operator<(const bwl& b) const{
        return l < b.l;
    }
};

bwl ee[];

int main() {
    srand(time());
    while (scanf("%d%d", &n, &m) != EOF) {
        TreapNode *root = NULL;
        for (int i = ; i < n; i++) {
            scanf("%d", &a[i]);
        }
        for (int i = ; i < m; i++) {
            scanf("%d%d%d", &ee[i].l, &ee[i].r, &ee[i].k);
            ee[i].idx = i;
        }
        sort(ee, ee + m);
        int pre = ,last = ;
        for (int i = ; i < m; i++) {
            if (i == ) {
                pre = last = ee[i].l;
            }
            while (pre <= last && pre < ee[i].l) {
                Delete(root, a[pre-]);
                pre++;
            }
            while (last <= n && last <= ee[i].r) {
                Insert(root, a[last-]);
                last++;
            }
            ans[ee[i].idx] = Select(root, ee[i].k);
        }
        for (int i = ; i < m; i++) {
            printf("%d\n", ans[i]);
        }
        delete root;
    }
    return ;
}

/*
 7 2
 1 5 2 6 3 7 4
 1 5 3
 2 7 1

 9 3
 1 3 10 2 4 7 7 8 2
 1 3 3
 5 9 2
 3 8 4
 */