【BZOJ3932】任务查询系统（主席树）
【BZOJ3923】任务查询系统（主席树）
题面
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的一个排列
题解
真是一道好题
对于一个任务出现的时间
拆为：S时刻出现，E+1时刻消失
那么，以优先级为权值，构建主席树
把任务拆开后，很容易利用主席树求前缀和
所以第K大也就很容易的求出来了
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 120000
#define ll long long
inline int read()
{
int x=0,t=1;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
struct Event
{
int t,p,w;
}e[MAX<<2];
struct Node
{
int ls,rs;
int v;
long long s;
}t[MAX<<5];
int n,m,tot,sum,S[MAX],cnt,rt[MAX];
bool operator<(Event a,Event b){return a.t<b.t;}
void Build(int &now,int l,int r)
{
now=++tot;
if(l==r)return;
int mid=(l+r)>>1;
Build(t[now].ls,l,mid);
Build(t[now].rs,mid+1,r);
}
void Modify(int &now,int ff,int l,int r,int pos,int w)
{
now=++tot;
t[now]=t[ff];t[now].v+=w;t[now].s+=S[pos]*w;
if(l==r)return;
int mid=(l+r)>>1;
if(pos<=mid)Modify(t[now].ls,t[ff].ls,l,mid,pos,w);
else Modify(t[now].rs,t[ff].rs,mid+1,r,pos,w);
}
ll Query(int now,int l,int r,int k)
{
if(t[now].v<=k)return t[now].s;
if(l==r)return t[now].s/(1ll*t[now].v)*1ll*k;
int mid=(l+r)>>1;
if(k<=t[t[now].ls].v)return Query(t[now].ls,l,mid,k);
else return t[t[now].ls].s+Query(t[now].rs,mid+1,r,k-t[t[now].ls].v);
}
int main()
{
m=read();n=read();
for(int i=1;i<=m;++i)
{
cnt++;
e[cnt].t=read();e[cnt+1].t=read()+1;
S[++sum]=e[cnt].p=e[cnt+1].p=read();
e[cnt].w=1;e[cnt+1].w=-1;
cnt++;
}
sort(&S[1],&S[sum+1]);
sum=unique(&S[1],&S[sum+1])-S-1;
sort(&e[1],&e[cnt+1]);
Build(rt[0],1,sum);
for(int i=1;i<=cnt;++i)
{
int pp=lower_bound(&S[1],&S[sum+1],e[i].p)-S;
Modify(rt[e[i].t],rt[e[i-1].t],1,sum,pp,e[i].w);
}
for(int i=1;i<=n;++i)
if(!rt[i])rt[i]=rt[i-1];
ll ls=1;
while(n--)
{
int X=read(),A=read(),B=read(),C=read();
int K=1+(1ll*A*ls%C+B)%C;
printf("%lld\n",ls=Query(rt[X],1,sum,K));
}
return 0;
}
