BZOJ3932 CQOI2015 任务查询系统 【主席树】

BZOJ3932 CQOI2015 任务查询系统

---

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的一个排列

---

主席树板子题，我们把一个时间段拆成差分的形式，然后可以用vector存一下每个时间点有哪些加入、删除。建树的时候先把rt[0]所对应的树完整建出来，然后对于每一个事件直接继承上一个时间然后在这时间对应的树修改就好了

---

#include<bits/stdc++.h>
using namespace std;
#define N 4000010
#define LL long long
struct Node{LL vl,typ;};
LL cnt=0,tot=0,maxt=0,pre=1,b[N];
LL rt[N],ls[N],rs[N],siz[N],sum[N];
vector<Node> G[N];
map<LL,LL> mp;
LL read(){
    LL ans=0,w=1;char c=getchar();
    while(!isdigit(c)&&c!='-')c=getchar();
    if(c=='-')c=getchar(),w=-1;
    while(isdigit(c))ans=ans*10+c-'0',c=getchar();
    return ans*w;
}
void build(LL &t,LL l,LL r){
    if(l>r)return;
    t=++cnt;
    if(l==r)return;
    LL mid=(l+r)>>1;
    build(ls[t],l,mid);
    build(rs[t],mid+1,r); 
}
void modify(LL &t,LL last,LL l,LL r,LL vl,LL pos,LL typ){
    t=++cnt;
    sum[t]=sum[last]+vl;
    siz[t]=siz[last]+typ;
    ls[t]=ls[last];
    rs[t]=rs[last];
    if(l==r)return;
    LL mid=(l+r)>>1;
    if(pos<=mid)modify(ls[t],ls[last],l,mid,vl,pos,typ);
    else modify(rs[t],rs[last],mid+1,r,vl,pos,typ);
}
LL query(LL t,LL l,LL r,LL k){
    if(k>=siz[t])return sum[t];
    if(l==r)return sum[t]/siz[t]*k;
    LL mid=(l+r)>>1,tmp=siz[ls[t]];
    if(k<=tmp)return query(ls[t],l,mid,k);
    else return query(rs[t],mid+1,r,k-tmp)+sum[ls[t]]; 
}
int main(){
    LL n=read(),m=read();
    for(LL i=1;i<=n;i++){
        LL s=read(),e=read(),p=read();
        b[i]=p;
        G[s].push_back((Node){p,1});
        G[e+1].push_back((Node){-p,-1});
        maxt=max(maxt,e+1);
    }
    sort(b+1,b+n+1);
    for(LL i=1;i<=n;i++)if(!mp[b[i]])mp[b[i]]=++tot;
    build(rt[0],1,tot);
    for(LL i=1;i<=maxt;i++){
        rt[i]=rt[i-1];
        for(LL j=0;j<G[i].size();j++)
            modify(rt[i],rt[i],1,tot,G[i][j].vl,mp[labs(G[i][j].vl)],G[i][j].typ);
    }
    for(LL i=1;i<=m;i++){
        LL x=read(),a=read(),b=read(),c=read(),k=(a*pre+b)%c+1;
        printf("%lld\n",pre=query(rt[x],1,tot,k));
    }
    return 0;
}