B -- RE：从零开始的异世界生活 线段树

http://www.ifrog.cc/acm/problem/1117?contest=1016&no=1

其实我是第一次这样用线段树。

首先把所有出现过的数字全部离散化。那么数字就是从[1,  100000]了。

把他们看成一个个独立的节点。每次插入一个数字，就在那个位置加1.

那么线段树维护区间总和，想知道小于x的数字有多少个，直接区间查询[1, x - 1]即可。

如果把小于x的数，变成x，相当于把[1, x - 1]中出现了多少个数字，贡献 add 到 x这个位置，然后把[1, x - 1]清0即可。

查询第k小，可以二分然后判断。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <assert.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <bitset>
#define lson L, mid, cur << 1
#define rson mid + 1, R, cur << 1 | 1
const int maxn =  + ;
struct Node {
    int op, val;
}query[maxn];
vector<int>vc;
set<int>ss;
int sum[maxn << ];
int add[maxn << ];
void pushUp(int cur) {
    sum[cur] = sum[cur << ] + sum[cur <<  | ];
}
void pushDown(int cur, int total) {
    if (add[cur] != -inf) {
        add[cur << ] = add[cur];
        add[cur <<  | ] = add[cur];
        sum[cur << ] = sum[cur <<  | ] = ;
        add[cur] = -inf;
    }
}
void upDate(int be, int en, int val, int L, int R, int cur) {
    if (be > en) return;
    if (L >= be && R <= en) {
        add[cur] = val;
        sum[cur] = val;
        return;
    }
    pushDown(cur, R - L + );
    int mid = (L + R) >> ;
    if (be <= mid) upDate(be, en, val, lson);
    if (en > mid) upDate(be, en, val, rson);
    pushUp(cur);
}
int ask(int be, int en, int L, int R, int cur) {
    if (en < be) return ;
    if (L >= be && R <= en) return sum[cur];
    pushDown(cur, R - L + );
    int ans = , mid = (L + R) >> ;
    if (be <= mid) ans += ask(be, en, lson);
    if (en > mid) ans += ask(be, en, rson);
    return ans;
}
#define root 1, n, 1
int n;
int slove(int x) { //要大于多少个number
    int be = , en = n;
    while (be <= en) {
        int mid = (be + en) >> ;
        int res = ask(, mid, root);
        if (res >= x) en = mid - ;
        else be = mid + ;
    }
    return vc[be];
}
void work() {
    int en = maxn << ;
    for (int i = ; i <= en - ; ++i) add[i] = -inf;
    vc.push_back(-inf);
    int q;
    scanf("%d", &q);
    for (int i = ; i <= q; ++i) {
        scanf("%d%d", &query[i].op, &query[i].val);
        ss.insert(query[i].val);
    }
    for (set<int> :: iterator it = ss.begin(); it != ss.end(); ++it) {
        vc.push_back(*it);
    }
    n = vc.size();
    n--;
    int nowHas = ;
    for (int i = ; i <= q; ++i) {
        int op = query[i].op, val = query[i].val;
        int pos = lower_bound(vc.begin(), vc.end(), val) - vc.begin();
        if (op == ) {
            nowHas++;
            int has = ask(pos, pos, root);
            upDate(pos, pos, has + , root);
        } else if (op == ) {
            int res = ask(, pos, root);
            upDate(pos, pos, res, root);
            upDate(, pos - , , root);
        } else if (op == ) {
            int res = ask(, n, root) - ask(, pos, root);
            upDate(pos, pos, res, root);
            upDate(pos + , n, , root);
        } else if (op == ) {
            printf("%d\n", slove(val));
        } else if (op == ) {
            int res = ask(, pos - , root);
            printf("%d\n", res);
        }
    }
//    printf("*****************\n");
//    printf("%d\n", ask(1, 3, root));
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    work();
    return ;
}