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

luogu & bzoj

题目描述

最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的任务用三元组(Si,Ei,Pi)描述,(Si,Ei,Pi)表示任务从第Si秒开始,在第Ei秒后结束(第Si秒和Ei秒任务也在运行),其优先级为Pi。同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同。调度系统会经常向查询系统询问,第Xi秒正在运行的任务中,优先级最小的Ki个任务(即将任务按照优先级从小到大排序后取前Ki个)的优先级之和是多少。特别的,如果Ki大于第Xi秒正在运行的任务总数,则直接回答第Xi秒正在运行的任务优先级之和。上述所有参数均为整数,时间的范围在1到n之间(包含1和n)。

输入输出格式

输入格式:

输入文件第一行包含两个空格分开的正整数m和n,分别表示任务总数和时间范围。接下来m行,每行包含三个空格分开的正整数Si、Ei和Pi(Si<=Ei),描述一个任务。接下来n行,每行包含四个空格分开的整数Xi、Ai、Bi和Ci,描述一次查询。查询的参数Ki需要由公式 Ki=1+(Ai*Pre+Bi) mod Ci计算得到。其中Pre表示上一次查询的结果,对于第一次查询,Pre=1。

输出格式:

输出共n行,每行一个整数,表示查询结果。

解题思路

代码

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #include<cstring>
 5 #include<math.h>
 6 #define LL long long
 7 using namespace std;
 8 const int maxn=(1e5+10)*85;
 9 struct nod{
10     int pos,val;
11 };
12 nod p[maxn];
13 LL root[maxn];
14 LL sum[maxn],num[maxn],lson[maxn],rson[maxn];
15 int end[maxn],num_start[maxn];
16 int n,m,cntt=0,cnt=1;
17 inline bool cmp(nod a,nod b){
18     return a.pos<b.pos;
19 }
20 inline void read(long long &x){
21     x=0; register char ch=getchar();
22     while(ch<'0'||ch>'9')ch=getchar();
23     while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
24 }
25 inline int ins(LL &ID,LL IDl,int nl,int nr,int val){
26     if(ID==0)ID=++cnt;
27     if(val>0)num[ID]=num[IDl]+1;
28     else num[ID]=num[IDl]-1;
29     sum[ID]=sum[IDl]+val;
30     if(nl==nr)return 0;
31     int m=(nl+nr)>>1;
32     if(m>=abs(val)){
33         rson[ID]=rson[IDl];
34         ins(lson[ID],lson[IDl],nl,m,val);
35     }
36     else {
37         lson[ID]=lson[IDl];
38         ins(rson[ID],rson[IDl],m+1,nr,val);
39     }
40     num[ID]=num[lson[ID]]+num[rson[ID]];
41     sum[ID]=sum[lson[ID]]+sum[rson[ID]];
42 }
43 inline LL query(int nl,int nr,int now,int k){
44     if(nl==nr)return (LL)nl*k;
45     int m=(nl+nr)>>1;
46     if(num[lson[now]]>=k)return query(nl,m,lson[now],k);
47     else return query(m+1,nr,rson[now],k-num[lson[now]])+sum[lson[now]];
48 }
49 int main(){
50     int max_val=-99999999;
51     scanf("%d%d",&m,&n);
52     register int S,E,P;
53     for(register int i=1;i<=m;i++){
54         scanf("%d%d%d",&S,&E,&P);
55         p[++cntt].pos=S; p[cntt].val=P;
56         p[++cntt].pos=E+1; p[cntt].val=-P;
57         max_val=max(max_val,P);
58     }
59     sort(p+1,p+cntt+1,cmp);
60     root[0]=1;
61     for(register int i=1;i<=cntt;i++)
62         ins(root[i],root[i-1],1,max_val,p[i].val);
63     for(register int i=cntt;i>=0;i--)
64         if(p[i].pos!=p[i+1].pos)end[p[i].pos]=i;
65     for(register int i=1;i<=1e6;i++)
66         if(!end[i])end[i]=end[i-1];
67     register long long X,A,B,C,K,pre=1;
68     for(register int i=1;i<=n;i++){
69         read(X),read(A),read(B),read(C);
70         K=1+(A*pre+B)%C;
71         if(num[root[end[X]]]<=K)pre=sum[root[end[X]]];
72         else pre=query(1,max_val,root[end[X]],K);
73         printf("%lld\n",pre);
74     }
75 }

猜你喜欢

转载自www.cnblogs.com/Fang-Hao/p/8961818.html