BZOJ 3932: [CQOI2015]任务查询系统(主席树)
BZOJ 3932: [CQOI2015]任务查询系统(主席树)
Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 4509 Solved: 1497
Description
最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的
任务用三元组(Si,Ei,Pi)描述，(Si,Ei,Pi)表示任务从第Si秒开始，在第Ei秒后结束（第Si秒和Ei秒任务也在运行
），其优先级为Pi。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。调度系统会经常向
查询系统询问，第Xi秒正在运行的任务中，优先级最小的Ki个任务（即将任务按照优先级从小到大排序后取前Ki个
）的优先级之和是多少。特别的，如果Ki大于第Xi秒正在运行的任务总数，则直接回答第Xi秒正在运行的任务优先
级之和。上述所有参数均为整数，时间的范围在1到n之间（包含1和n）。
Input
输入文件第一行包含两个空格分开的正整数m和n，分别表示任务总数和时间范围。接下来m行，每行包含三个空格
分开的正整数Si、Ei和Pi(Si≤Ei)，描述一个任务。接下来n行，每行包含四个空格分开的整数Xi、Ai、Bi和Ci，
描述一次查询。查询的参数Ki需要由公式 Ki=1+(Ai*Pre+Bi) mod Ci计算得到。其中Pre表示上一次查询的结果，
对于第一次查询，Pre=1。
Output
输出共n行，每行一个整数，表示查询结果。
Sample Input
4 3 1 2 6 2 3 3 1 3 2 3 3 4 3 1 3 2 1 1 3 4 2 2 4 3
Sample Output
2 8 11
HINT
样例解释
K1 = (1*1+3)%2+1 = 1
K2 = (1*2+3)%4+1 = 2
K3 = (2*8+4)%3+1 = 3
对于100%的数据，1≤m,n,Si,Ei,Ci≤100000，0≤Ai,Bi≤100000，1≤Pi≤10000000，Xi为1到n的一个排列
Source
很水但是很恶心
首先不难想到对两个端点进行差分
然后把主席树建出来就可以了
注意要开ll
另外，叶子节点的总权值可能大于查询的个数，需要特判
#include<cstdio>
#include<algorithm>
#include<vector>
#define int long long
using namespace std;
const int MAXN = 1e6 + 10;
inline int read() {
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
int N, M, PRE = 1;
struct node {
int ls, rs, sum, siz;
}T[MAXN * 6];
int root[MAXN], tot = 0, realrt[MAXN];
int S[MAXN], E[MAXN], tmp[MAXN], P[MAXN];
vector<int>v[MAXN];
#define rs(x) T[x].rs
#define ls(x) T[x].ls
int insert(int &now, int pre, int l, int r, int pos, int val) {
now = ++tot;
T[now] = T[pre]; T[now].sum += val;
val > 0 ? T[now].siz++ : T[now].siz--;
if(l == r) return now;
int mid = l + r >> 1;
if(pos <= mid) ls(now) = insert(ls(now), ls(pre), l, mid, pos, val);
else rs(now) = insert(rs(now), rs(pre), mid + 1, r, pos, val);
return now;
}
int Query(int now, int l, int r, int val) {
if(l == r) return min(val * tmp[l], T[now].sum);
int mid = l + r >> 1;
if(val <= T[ls(now)].siz) return Query(ls(now), l, mid, val);
else return Query(rs(now), mid + 1, r, val - T[ls(now)].siz) + T[ls(now)].sum;
}
main() {
#ifdef WIN32
freopen("a.in", "r", stdin);
freopen("a.out", "w", stdout);
#endif
N = read(); M = read();
for(int i = 1; i <= N; i++) {
S[i] = read(), E[i] = read(), tmp[i] = P[i] = read();
}
sort(tmp + 1, tmp + N + 1);
int num = unique(tmp + 1, tmp + N + 1) - tmp - 1;
for(int i = 1; i <= N; i++) {
P[i] = lower_bound(tmp + 1, tmp + num + 1, P[i]) - tmp;
v[S[i]].push_back(P[i]); v[E[i] + 1].push_back(-P[i]);
}
int cur = 0;
for(int i = 1; i <= 100000; i++) {
for(int j = 0; j < v[i].size(); j++)
cur++, insert(root[cur], root[cur - 1], 1, num, abs(v[i][j]), v[i][j] < 0 ? -tmp[-v[i][j]] : tmp[v[i][j]]);
realrt[i] = cur;
}
for(int i = 1; i <= M; i++) {
int Xi = read(), Ai = read(), Bi = read(), Ci = read();
int Ki = 1 + (Ai * PRE + Bi) % Ci;
PRE = Query(root[realrt[Xi]], 1, num, Ki);
printf("%lld\n", PRE);
}
}