[BZOJ2028][SHOI2009]会场预约(Splay)

考虑以日期为关键字,把预约用Splay进行维护。
先考虑加入新的预约。设新的预约是从l日到r日。
很显然,一个从l0日到r0日的预约与这个新的预约不冲突的充分必要条件为:r0<ll0>r
首先找出满足r0<l的情况下r0最大的节点x,以及满足l0>r的情况下l0最小的节点y。这显然就是平衡树中求前驱后继的操作。
下面使用类似于Splay维护序列时提取区间的操作,把节点xSplay到根,再把ySplay到根的右子节点。那么y的左子树就包含了所有冲突的预约。此时冲突的预约个数就是y的左子树的size。此时,就可以新建一个节点(新的预约),并用这个节点替换掉y的左子树,并维护下size
2个操作就是询问根节点的size(我的程序稍微写麻烦了点)。
代码:

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
inline char get() {
    char c; while ((c = getchar()) != 'A' && c != 'B'); return c;
}
const int N = 2e5 + 5;
int n, T, Rt, cnt, fa[N], lc[N], rc[N], sze[N], L[N], R[N];
int which(int x) {return rc[fa[x]] == x;}
void upt(int x) {
    sze[x] = 1;
    if (lc[x]) sze[x] += sze[lc[x]];
    if (rc[x]) sze[x] += sze[rc[x]];
}
void rotate(int x) {
    int y = fa[x], z = fa[y], b = lc[y] == x ? rc[x] : lc[x];
    if (z) (lc[z] == y ? lc[z] : rc[z]) = x;
    fa[x] = z; fa[y] = x; if (b) fa[b] = y;
    if (lc[y] == x) rc[x] = y, lc[y] = b;
    else lc[x] = y, rc[y] = b; upt(y); upt(x);
}
void splay(int x, int tar) {
    while (fa[x] != tar) {
        if (fa[fa[x]] != tar) {
            if (which(x) == which(fa[x])) rotate(fa[x]);
            else rotate(x);
        }
        rotate(x);
    }
    if (!tar) Rt = x;
    upt(x);
}
int PreR(int v) {
    int x = Rt, res = 1;
    while (x)
        if (R[x] < v) res = x, x = rc[x];
        else x = lc[x];
    return res;
}
int SufL(int v) {
    int x = Rt, res = 2;
    while (x)
        if (L[x] > v) res = x, x = lc[x];
        else x = rc[x];
    return res;
}
void ins(int l, int r) {
    int x = PreR(l), y = SufL(r);
    splay(x, 0); splay(y, Rt);
    printf("%d\n", (lc[rc[Rt]] ? sze[lc[rc[Rt]]] : 0));
    if (lc[rc[Rt]]) {
        cnt -= sze[lc[rc[Rt]]];
        fa[lc[rc[Rt]]] = 0; lc[rc[Rt]] = 0;
        upt(rc[Rt]); upt(Rt);
    }
    L[lc[rc[Rt]] = ++T] = l; R[T] = r; fa[T] = rc[Rt];
    sze[T] = 1; upt(rc[Rt]); upt(Rt); cnt++;
}
void init() {
    fa[T = rc[Rt = 1] = 2] = sze[2] = 1;
    L[sze[1] = 2] = R[2] = 1e5 + 3;
}
int main() {
    int i, x, y; n = read(); char c; init();
    while (n--) {
        c = get(); if (c == 'A') {
            x = read(); y = read();
            ins(x, y);
        }
        else printf("%d\n", cnt);
    }
    return 0;
}
发布了410 篇原创文章 · 获赞 53 · 访问量 7万+
展开阅读全文

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

©️2019 CSDN 皮肤主题: 创作都市 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览