bzoj3065: 带插入区间K小值

无聊来写了下

一开始发现树高是O(n)的，然后就MLE了，进去看了下发现没有重构！

看了半天发现调用错了函数

然后进去又发现不满足sz = ch[0]->sz + ch[1]->sz + 1

然而全都是maintain过的啊

发现后来受某个代码的影响

返回重建节点时把引用去掉了

这样这个点的父亲的信息就不对了

又仔细去看了一下那个人为什么这样做没问题

结果他每次都把内存回收了，而且重建的根节点是最后删除第一个拿出来的（然后他竟然用的队列，当我什么都没说，我还是不知道为什么是对的。。）

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>
//#include<cassert>

using namespace std;

#define ALL_VALUE 0, 70000
const int N = 70000 + 10;

int CNT1 = 0, CNT2 = 0;

namespace ST {
    struct Node *stk[N * 60]; int top;
    struct Node {
        int sz;
        Node* ch[2];
        
        Node() {}
        Node(int sz) : sz(sz) {
            ch[0] = ch[1] = 0;
        }
        
        void *operator new (size_t) {
//            if(!++CNT1 % 100000) cerr << CNT1 << endl;
            if(top) return stk[top--];
            static Node *pis;
            static int left = 0;
            if(!left) left = N, pis = new Node[N] ();
            return left--, pis++;
        }
        
        void operator delete(void *p) {
//            ++CNT2;
            stk[++top] = (Node*) p;
        }
    };
    
    #define mid ((l + r) >> 1)
    #define ls o->ch[0], l, mid
    #define rs o->ch[1], mid + 1, r
    
    void insert(Node*& o, int l, int r, int v, int d) {
        if(!o) o = new Node(0);
        o->sz += d;
        if(l == r) return;
        if(v <= mid) insert(ls, v, d);
        else insert(rs, v, d);
        if(!o->sz) delete o, o = 0;
    }
    
    int query(Node* o, int l, int r, int v) {// <=v µÄÊýµÄ¸öÊý 
        if(!o) return 0;
        if(r <= v) return o->sz;
        return query(ls, v) + (v > mid ? query(rs, v) : 0);
    }
    
    void remove(Node*& o) {
        if(!o) return;
        remove(o->ch[0]);
        remove(o->ch[1]);
        delete o, o = 0;
    }
    
    #undef mid
    #undef ls
    #undef rs
}

int a[N], dep = 0;

namespace SGT {
    struct Node *pis;
    struct Node {
        int sz, v;
        
        Node *ch[2];
        ST::Node *da;
        
        int cmp(int k) const {
            int s = ch[0]->size() + 1;
            if(s == k) return -1;
            return k < s ? 0 : 1;
        }
        
        Node(int v = 0) : v(v) {
            ch[0] = ch[1] = 0;
            da = 0;
            sz = 1;
        }
        
        int size() const {
            return this ? sz : 0;
        }
        
        void maintain() {
            sz = ch[0]->size() + ch[1]->size() + 1;
        }
        
        int query(int k, int v) const {//²éѯ1-kÀï <= vµÄÊýµÄÊýÁ¿
            if(!this || !k) return 0;
//            cerr << ch[0]->size() << endl;
            if(ch[0]->size() >= k) return ch[0]->query(k, v);
            return (ch[0] ? ST::query(ch[0]->da, ALL_VALUE, v) : 0) + (this->v <= v) + ch[1]->query(k - ch[0]->size() - 1, v);
        }
        
        void *operator new(size_t) {
            return pis++;
        }
    }*null, *root, pool[N];
    
    Node *rec[N]; int tot;
    
    void print(Node *o) {
        if(!o) return;
        print(o->ch[0]);
        ST::remove(o->da);
        rec[++tot] = o;
        print(o->ch[1]);
    }
    
    /*void check(Node *o) {
        if(!o) return;
        assert(o->sz == o->ch[0]->size() + o->ch[1]->size() + 1);
        check(o->ch[0]);
        check(o->ch[1]);
    }*/
    
    void rebuild(Node*& o, int l, int r) {
        if(l > r) return o = 0, void();
        int mid = (l + r) >> 1;
        o = rec[mid];
        for(int i = l; i <= r; i++) ST::insert(o->da, ALL_VALUE, rec[i]->v, 1);
        rebuild(o->ch[0], l, mid - 1);
        rebuild(o->ch[1], mid + 1, r);
        o->maintain();
//        assert(o->sz == o->ch[0]->size() + o->ch[1]->size() + 1);
    }
    
    void build(Node*& o, int l, int r) {
        if(l > r) return;
        int mid = (l + r) >> 1;
        o = new Node(a[mid]);
        for(int i = l; i <= r; i++) ST::insert(o->da, ALL_VALUE, a[i], 1);
        build(o->ch[0], l, mid - 1);
        build(o->ch[1], mid + 1, r);
        o->maintain();
    }
    
    Node*& insert(Node*& o, int k, int x) { // ÔÚµÚx¸öλÖúó²åÈëx 
        if(!o) {
            o = new Node(x);
            ST::insert(o->da, ALL_VALUE, x, 1);
            return null;
        }
//        assert(o->sz == o->ch[0]->size() + o->ch[1]->size() + 1);
        int d = o->ch[0]->size() >= k ? 0 : 1;
        if(d == 1) k -= o->ch[0]->size() + 1;
        Node*& res = insert(o->ch[d], k, x);
        o->maintain();
        ST::insert(o->da, ALL_VALUE, x, 1);
        if(o->ch[d]->size() > o->size() * 0.75) return o;
        else return res;
//        assert(o->sz == o->ch[0]->size() + o->ch[1]->size() + 1);
    }
    
    int modify(Node*& o, int k, int x) {
//        assert(o->sz == o->ch[0]->size() + o->ch[1]->size() + 1);
        int d = o->cmp(k), old;
        if(d == 1) k -= o->ch[0]->size() + 1;
        if(d == -1) old = o->v, o->v = x;
        else old = modify(o->ch[d], k, x);
        ST::insert(o->da, ALL_VALUE, old, -1);
        ST::insert(o->da, ALL_VALUE, x, 1);
        return old;
    }
    
    void insert(int k, int x) {
//        SGT::check(SGT::root);
        Node*& res = insert(root, k, x);
//        SGT::check(SGT::root);
        if(res != null) {
            tot = 0;
            print(res);
            rebuild(res, 1, tot);
//            SGT::check(SGT::root);
        }
    }
}

int query(int l, int r, int k) {
    int L = 0, R = 70000, res = -1;
    while(L <= R) {
        int mid = (L + R) >> 1;
        int t1 = SGT::root->query(r, mid);
        int t2 = SGT::root->query(l - 1, mid);
        if(t1 - t2 >= k) R = mid - 1, res = mid;
        else L = mid + 1;
    }
    return res;
}

int main() {
#ifdef DEBUG
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    
    SGT::pis = SGT::pool;
    
    int x, n, m;
    scanf("%d", &n);
    
    for(int i = 1; i <= n; i++) {
        scanf("%d", a + i);
    }
    
    int la = 0;
    
    SGT::build(SGT::root, 1, n);
    
    scanf("%d", &m);
    char opt[10]; int y, k, v;
    
    for(int i = 1; i <= m; i++) {
//        if(i == 2859) 
//        int debug = 1;
//        SGT::check(SGT::root);
        
        scanf("%s", opt);
        if(opt[0] == 'Q') {
            scanf("%d%d%d", &x, &y, &k);
            x ^= la, y ^= la, k ^= la;
            printf("%d
", query(x, y, k));
        }else if(opt[0] == 'M') {
            scanf("%d%d", &x, &v);
            x ^= la, v ^= la;
            SGT::modify(SGT::root, x, v);
        }else {
            scanf("%d%d", &x, &v);
            x ^= la, v ^= la;
//            int last = CNT1; dep = 0;
            SGT::insert(x - 1, v);
//            cerr << SGT::root->sz << endl;
//            cerr << CNT1 - last << ' ' << dep << endl;
        }
    }
    
    return 0;
}