[bzoj3932][CQOI2015]任务查询系统_主席树
任务查询系统 bzoj-3932 CQOI-2015
题目大意:最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的任务用三元组(Si,Ei,Pi)描述,(Si,Ei,Pi)表示任务从第Si秒开始,在第Ei秒后结束(第Si秒和Ei秒任务也在运行),其优先级为Pi。同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同。调度系统会经常向查询系统询问,第Xi秒正在运行的任务中,优先级最小的Ki个任务(即将任务按照优先级从小到大排序后取前Ki个)的优先级之和是多少。特别的,如果Ki大于第Xi秒正在运行的任务总数,则直接回答第Xi秒正在运行的任务优先级之和。上述所有参数均为整数,时间的范围在1到n之间(包含1和n)。
注释:$1\le n,m\le 10^5$。
想法:离散化之后lower_bound,主席树查询即可。
最后,附上丑陋的代码... ...
#include <queue> #include <cmath> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; int n,m,num; int tot,len,x,y,z; long long pre,a,b,c; int s[5000010],l[5000010],r[5000010],root[100010],cnt[5000010]; long long sum[5000010]; vector<int>p[100010],q[100010]; void build(int &x,int y) { x=++tot; l[x]=l[y]; r[x]=r[y]; sum[x]=sum[y]; cnt[x]=cnt[y]; s[x]=num; } void updata(int L,int R,int x,int &y,int v,int c) { if(s[x]!=num) { build(y,x); } cnt[y]+=c; sum[y]+=1ll*c*v; if(L==R) { return ; } if(v<=mid) { updata(L,mid,l[x],l[y],v,c); } else { updata(mid+1,R,r[x],r[y],v,c); } } long long query(int L,int R,int x,int k) { if(L==R) { return 1ll*L*k; } if(cnt[l[x]]>=k) { return query(L,mid,l[x],k); } else { return query(mid+1,R,r[x],k-cnt[l[x]])+sum[l[x]]; } } int main() { scanf("%d%d",&n,&m); int mn=1e9; int mx=-1e9; for(int i=1;i<=n;i++) { scanf("%d%d%d",&x,&y,&z); mn=min(mn,x); mx=max(mx,y); p[x].push_back(z); q[y].push_back(z); } for(int i=mn;i<=mx;i++) { build(root[i],root[i-1]); num++; len=p[i].size(); for(int j=0;j<len;j++) { updata(1,1e7,root[i],root[i],p[i][j],1); } len=q[i-1].size(); for(int j=0;j<len;j++) { updata(1,1e7,root[i],root[i],q[i-1][j],-1); } } pre=1; for(int i=1;i<=m;i++) { scanf("%d%lld%lld%lld",&x,&a,&b,&c); int k=(int) 1+(a*pre+b)%c; if(cnt[root[x]]<=k) { pre=sum[root[x]]; printf("%lld\n",pre); } else { pre=query(1,1e7,root[x],k); printf("%lld\n",pre); } } }
小结:主席树好好玩... ...
| 欢迎来原网站坐坐! >原文链接<