[CQOI2015]任务查询系统 主席树

[CQOI2015]任务查询系统

LG传送门

以前还没见过主席树的这种写法。

考虑使用差分的思想处理每一个任务，然后所有的东西就都能顺理成章地用主席树维护了，查询的时候和平时的主席树有一点不同，详见代码。

关于为什么可以做的一点思考：主席树其实类似于求前缀和，平时我们是用来区间查询，这里对于差分后的信息用主席树一维护，就变成了原来的信息，可以单点查询，而我们这道题的要求正好是区间修改单点查询。

//written by newbiechd
#include <cstdio>
#include <cctype>
#include <algorithm>
#define R register
#define I inline
#define B 1000000
#define L long long
using namespace std;
const int N = 100003, M = 1e7;
char buf[B], *p1, *p2;
I char gc() { return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, B, stdin), p1 == p2) ? EOF : *p1++; }
I int rd() {
    R int f = 0;
    R char c = gc();
    while (c < 48 || c > 57)
        c = gc();
    while (c > 47 && c < 58)
        f = f * 10 + (c ^ 48), c = gc();
    return f;
}
int rt[N], T;
struct task {
    int t, p, d;
    task () {}
    task (int t, int p, int d) : t(t), p(p), d(d) {}
    I int operator < (const task &x) { return t < x.t; }
}t[N << 1];
struct segtree {
    int p, q, s;
    L t;
}e[N << 6];
int insert(int k, int l, int r, int x, int y) {
    R int t = ++T;
    e[t] = e[k], e[t].s += y, e[t].t += y * x;
    if (l == r)
        return t;
    R int m = (l + r) >> 1;
    if (x <= m)
        e[t].p = insert(e[k].p, l, m, x, y);
    else
        e[t].q = insert(e[k].q, m + 1, r, x, y);
    return t;
}
L query(int k, int l, int r, int x) {
    if (l == r)
        return l * x;
    R int m = (l + r) >> 1, t = e[e[k].p].s;
    if (x <= t)
        return query(e[k].p, l, m, x);
    else
        return e[e[k].p].t + query(e[k].q, m + 1, r, x - t);
}
int main() {
    int m = rd(), n = rd(), S = 0, i, j, x, y, z;
    L ans = 1;
    for (i = 1; i <= m; ++i)
        x = rd(), y = rd(), z = rd(),
            t[++S] = task(x, z, 1), t[++S] = task(y + 1, z, -1);
    sort(t + 1, t + S + 1);
    for (i = 1, j = 1; i <= n; ++i) {
        rt[i] = rt[i - 1];
        for (; j <= S && t[j].t == i; ++j)
            rt[i] = insert(rt[i], 1, M, t[j].p, t[j].d);
    }
    for (i = 1; i <= n; ++i) {
        x = rd(), y = (ans * rd() + rd()) % rd() + 1;
        if (e[rt[x]].s <= y)
            ans = e[rt[x]].t;
        else
            ans = query(rt[x], 1, M, y);
        printf("%lld\n", ans);
    }
    return 0;
}