BZOJ3932[CQOI2015]任务查询系统——主席树
题目描述
最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的
任务用三元组(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行,每行一个整数,表示查询结果。
样例输入
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
样例输出
2
8
11
8
11
提示
样例解释
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的一个排列
这道题要求强制在线,因此用主席树。记录每一时刻的事件,对于一个区间[l,r]在l时刻+1,在r+1时刻-1。然后按时间顺序插入主席树,维护区间权值和及区间数的个数,对于每个询问查询对应时刻线段树。注意pre初始值是1不是0!
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | #include<map> #include<set> #include<queue> #include<cmath> #include<cstdio> #include<vector> #include<cstring> #include<iostream> #include<algorithm> #define mid (L+R)/2 using namespace std; int n,m; int num; int tot; int len; int x,y,z; long long pre; long long a,b,c; int s[5000010]; int l[5000010]; int r[5000010]; int root[100010]; int 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); } } } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步