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

3932: [CQOI2015]任务查询系统

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3777  Solved: 1215
[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的一个排列

将时间离散化成单个点，套主席树模板

#include <cstdio>#include <cstring>#include <algorithm>#include <bits/stdc++.h>using namespace std;const int N = 2e5+10;typedef long long LL;const LL mod = 1e9+7;struct node{    int x, v, num, id;    LL y;    bool operator <(const node &A)const    {        return x<A.x;    }}p[9000000];LL b[9000000], sum[9000000];int rt[9000000], tot;int ls[9000000], rs[9000000];void update(int &o,int last,int l,int r,int pos,int v){    o=++tot;    ls[o]=ls[last],rs[o]=rs[last];    sum[o]=sum[last]+v;    if(l==r) return ;    int mid=(l+r)/2;    if(pos<=mid)update(ls[o],ls[last],l,mid,pos,v);    else update(rs[o],rs[last],mid+1,r,pos,v);    return ;}LL query(int tt,int l,int r,LL num){    if(sum[tt]<num) num=sum[tt];    if(l==r) return b[l]*num;    LL tmp=sum[ls[tt]];    LL mid=(l+r)/2;    if(tmp>=num) return query(ls[tt],l,mid,num);    else return query(ls[tt],l,mid,tmp)+query(rs[tt],mid+1,r,num-tmp);}int main(){    int m, n;    scanf("%d %d", &m, &n);    int k=0;    for(int i=1;i<=m;i++)    {        int s, e;        LL px;        scanf("%d %d %lld", &s, &e, &px);        p[++k].x=s,p[k].v=1,p[k].y=px;        p[++k].x=e+1,p[k].v=-1,p[k].y=px;        b[i]=px;    }    sort(b+1,b+m+1);    int len=unique(b+1,b+m+1)-(b+1);    for(int i=1;i<=k;i+=2)    {        p[i].id=lower_bound(b+1,b+len+1,p[i].y)-b;        p[i+1].id=p[i].id;    }    sort(p+1,p+1+k);    memset(rt,0,sizeof(rt));    memset(sum,0,sizeof(sum));    tot=0;    int j=1;    for(int i=1;i<=n;i++)    {        rt[i]=rt[i-1];        while(p[j].x==i) update(rt[i],rt[i],1,len,p[j].id,p[j].v),j++;    }    LL ans=1;    for(int i=1;i<=n;i++)    {        LL x, ax, bx, c;        scanf("%lld %lld %lld %lld", &x, &ax, &bx, &c);        LL kx=1+(ax*ans+bx)%c;        ans=query(rt[x],1,len,kx);        printf("%lld\n",ans);    }    return 0;}