【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
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
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的一个排列
题解:本题是主席树的一个简化版吧,我们还是先离散化,然后按时间排序,每有一个任务开始或结束就新建一棵线段树,然后再查询时二分查找时间,然后再线段树上求前k个之和就行了,并且本题的主席树是不用相减的(这还叫主席树吗)
注意一下long long,注意二分边界不要搞错(尤其是用upper_bound的童鞋们),还要注意用离散化后的数来求出答案,还有第47行有点坑人
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; const ll maxn=200010; ll n,m,tot,nm,ans; struct task { ll pt,pv,pk; }p[maxn]; ll ls[40*maxn],rs[40*maxn],sum[40*maxn],siz[40*maxn]; ll rp[maxn>>1],root[maxn]; bool cmp1(task a,task b) { return a.pv<b.pv; } bool cmp2(task a,task b) { return a.pt<b.pt; } ll readin() { ll ret=0,f=1; char gc=getchar(); while(gc<'0'||gc>'9') {if(gc=='-')f=-f;gc=getchar();} while(gc>='0'&&gc<='9') ret=ret*10+gc-'0',gc=getchar(); return ret*f; } void insert(ll x,ll &y,ll l,ll r,ll pos,ll val) { y=++tot; if(l==r) { siz[y]=siz[x]+val; sum[y]=sum[x]+rp[l]*val; return ; } ll mid=l+r>>1; if(pos<=mid) rs[y]=rs[x],insert(ls[x],ls[y],l,mid,pos,val); else ls[y]=ls[x],insert(rs[x],rs[y],mid+1,r,pos,val); siz[y]=siz[ls[y]]+siz[rs[y]]; sum[y]=sum[ls[y]]+sum[rs[y]]; } ll query(ll x,ll l,ll r,ll pos) { if(l==r) return rp[l]*pos; ll mid=l+r>>1; if(siz[ls[x]]>=pos) return query(ls[x],l,mid,pos); else return sum[ls[x]]+query(rs[x],mid+1,r,pos-siz[ls[x]]); } int main() { n=readin(),m=readin(); ll i,a,b,c,d,e; for(i=1;i<=n;i++) p[i*2-1].pt=readin(),p[i*2].pt=readin()+1,p[i*2-1].pv=p[i*2].pv=readin(), p[i*2-1].pk=1,p[i*2].pk=-1; sort(p+1,p+2*n+1,cmp1); for(i=1;i<=2*n;i++) { if(rp[nm]<p[i].pv) rp[++nm]=p[i].pv; p[i].pv=nm; } sort(p+1,p+2*n+1,cmp2); for(i=1;i<=2*n;i++) insert(root[i-1],root[i],1,nm,p[i].pv,p[i].pk); ans=1; for(i=1;i<=m;i++) { d=readin(),a=readin(),b=readin(),c=readin(); e=1+(a*ans+b)%c; ll l=1,r=2*n+1,mid; while(l<r) { mid=l+r>>1; if(p[mid].pt<=d) l=mid+1; else r=mid; } l--; if(siz[root[l]]<=e) ans=sum[root[l]]; else ans=query(root[l],1,nm,e); printf("%lld\n",ans); } return 0; }
| 欢迎来原网站坐坐! >原文链接<