[CQOI2015]任务查询系统
以前还没见过主席树的这种写法。
考虑使用差分的思想处理每一个任务,然后所有的东西就都能顺理成章地用主席树维护了,查询的时候和平时的主席树有一点不同,详见代码。
关于为什么可以做的一点思考:主席树其实类似于求前缀和,平时我们是用来区间查询,这里对于差分后的信息用主席树一维护,就变成了原来的信息,可以单点查询,而我们这道题的要求正好是区间修改单点查询。
//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;
}