任务查询系统[主席树]

题目描述

最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的任务用三元组(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： 

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
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加一个值,r+1减一个值

主席树就可以做了

---

#include<bits/stdc++.h>
#define N 200005
#define LL long long
using namespace std;
int n,m,ret,tot,d,hash[N],rt[N];LL Pre=1;
struct Node{int pos,val,delta;}a[N];
struct Tree{int l,r,size;LL val;}t[N*20];
int read(){
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt*f;
}
bool cmp(Node a,Node b){return a.pos<b.pos;}
void build(int &o,int l,int r){
	o=++tot;
	if(l==r) return;
	int mid=l+r>>1;
	build(t[o].l,l,mid);
	build(t[o].r,mid+1,r);
}
void update(int &o,int l,int r,int pos,int pre,int delta){
	o=++tot;
	t[o].l=t[pre].l , t[o].r=t[pre].r;
	t[o].val = t[pre].val + (LL)(hash[pos] * delta);
	t[o].size = t[pre].size + delta;
	if(l==r) return;
	int mid=l+r>>1;
	if(pos<=mid) update(t[o].l , l , mid , pos , t[pre].l , delta);
	else update(t[o].r , mid+1 , r , pos , t[pre].r , delta);
}
LL quary(int o,int l,int r,int k){
	if(l==r) return (LL)t[o].val/t[o].size*k;
	int sl=t[t[o].l].size,mid=(l+r)>>1; LL ans=0;
	if(k<=sl) return quary(t[o].l,l,mid,k);
	else return t[t[o].l].val+quary(t[o].r,mid+1,r,k-sl);   
}
int main(){
	int m=read(),n=read();
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),z=read();
		a[++ret].pos=x,a[ret].val=z,a[ret].delta=1;
		a[++ret].pos=y+1,a[ret].val=z,a[ret].delta=-1;
		hash[i]=z;
	}
	sort(hash+1,hash+m+1);
	sort(a+1,a+ret+1,cmp);
	d=unique(hash+1,hash+m+1)-(hash+1);
	build(rt[0],1,d);
	for(int i=1;i<=ret;i++){
		int x=lower_bound(hash+1,hash+m+1,a[i].val)-hash;
		int now=a[i].pos;
		if(now==a[i-1].pos) update(rt[now] , 1 , d , x , rt[now] , a[i].delta);
		else{
			if(i!=1) for(int j=a[i-1].pos+1;j<now;j++) rt[j]=rt[j-1];
			update(rt[now] , 1 , d , x , rt[now-1] , a[i].delta);
		}
	}
	for(int i=1;i<=n;i++){
		int x=read(),A=read(),B=read(),C=read();
		LL k=1+(LL)(A*Pre+B)%C;
		if(k>=t[rt[x]].size) Pre=t[rt[x]].val;
		else Pre=quary(rt[x],1,d,k);
		printf("%lld\n",Pre);
	}
}