洛谷P3168 [CQOI2015] 任务查询系统 差分+主席树

给出$n\le 1e5$个任务的时间段$s_i,t_i,p_i$分别是这个任务的开始时间，结束时间以及优先级。然后给出一个时间点，求出当前时间点前$k$个任务的优先级之和。
求前$k$大用主席树维护，对每个时间点暴力插入时间复杂度过高，这里利用差分的思想，在$s_i$位置插入，而在$t_i+1$的位置删除。然后主席树本身维护就是前缀和。一次建树然后每次直接找到对应的根查询。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=2e5+7;
vector<int> T[N];
int b[N]; 
int ans[N]; 

int rt[N]; 
int ls[32*N],rs[32*N],cnt[32*N];
ll sum[32*N];
int tot=0;

void update(int &o,int pre,int l,int r,int x,int opt) {
	o=++tot;
	cnt[o]=cnt[pre]+opt;
	sum[o]=sum[pre]+1LL*opt*b[x];
	ls[o]=ls[pre];
	rs[o]=rs[pre];
	if(l==r) return;
	int mid=(l+r)>>1;
	if(x<=mid) update(ls[o],ls[pre],l,mid,x,opt);
	else update(rs[o],rs[pre],mid+1,r,x,opt);
}

ll query(int o,int l,int r,int k) {
	if(l==r) return b[l]*k;
	int mid=(l+r)>>1;
	if(k<=cnt[ls[o]]) return query(ls[o],l,mid,k);
	else return query(rs[o],mid+1,r,k-cnt[ls[o]])+sum[ls[o]];
}

int main() {
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) {
		int l,r,c;
		scanf("%d%d%d",&l,&r,&c);
		b[i]=c;
		T[l].push_back(c);
		T[r+1].push_back(-c);
	}
	sort(b+1,b+1+n);
	int sz=unique(b+1,b+1+n)-(b+1);
	int o=0; 
	for(int i=1;i<=m;i++) {
		int opt;
		for(int j=0;j<T[i].size();j++) {
			int c=T[i][j]; 
			if(c>0) opt=1;
			else { opt=-1,c=-c; }
			++o; 
			c=lower_bound(b+1,b+1+sz,c)-b;
			update(rt[o],rt[o-1],1,sz,c,opt);
		}
		ans[i]=o;
	}
	ll lastans=1;
	for(int i=1;i<=m;i++) {
		int x,a,b,c;
		scanf("%d%d%d%d",&x,&a,&b,&c);
		int k=1+(a*lastans+b)%c;
		x=rt[ans[x]]; 
		if(cnt[x]<k) {
			printf("%lld\n",sum[x]);
			lastans=sum[x];
		}
		else {
			lastans=query(x,1,sz,k);
			printf("%lld\n",lastans); 
		}
	}
	return 0;
}