题目描述
最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的任务用三元组(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行,每行一个整数,表示查询结果。
解题思路
当然,看到区间k大的时候自然而然就会想到主席树,而每一个任务是一个时间的区间,因为主席树自带差分的功能(在前面差分,后面的所有点不需要从头扫描一遍!!),所以,我们可以按照时间为根节点建立主席树,树上的每一个节点维护的是优先级,数据有当前区间内的个数和优先级的和。
代码
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 }
原文地址:https://www.cnblogs.com/Fang-Hao/p/8961818.html
时间: 2024-11-09 03:48:17