欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

BZOJ 3932: [CQOI2015]任务查询系统(主席树)

程序员文章站 2024-01-11 09:17:34
Description 最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的 任务用三元组(Si,Ei,Pi)描述,(Si,Ei,Pi)表示任务从第Si秒开始,在第Ei秒后结束(第Si秒和Ei秒任务也在运行 ),其优先级为Pi。同一时间可能有多个任务同 ......
Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 4509  Solved: 1497
[Submit][Status][Discuss]

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);
    }
}