BZOJ_3932_[CQOI2015]任务查询系统_主席树

BZOJ_3932_[CQOI2015]任务查询系统_主席树

题意:

最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分。超级计算机中的
任务用三元组(Si,Ei,Pi)描述,(Si,Ei,Pi)表示任务从第Si秒开始,在第Ei秒后结束(第Si秒和Ei秒任务也在运行
),其优先级为Pi。同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同。调度系统会经常向
查询系统询问,第Xi秒正在运行的任务中,优先级最小的Ki个任务(即将任务按照优先级从小到大排序后取前Ki个
)的优先级之和是多少。特别的,如果Ki大于第Xi秒正在运行的任务总数,则直接回答第Xi秒正在运行的任务优先
级之和。上述所有参数均为整数,时间的范围在1到n之间(包含1和n)。
 
分析:
对每秒的状态开一棵线段树
结点保存size和sum
主席树上差分直接搞
注意优先级最小的k个最后可能小于叶子节点的size
 
代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 800050
#define LL long long
const int maxn=10000000;
int t[N*20],ls[N*20],rs[N*20],n,m,cnt;
int head[N],to[N<<1],nxt[N<<1],val[N<<1],tot=1,root[N];
LL sum[N*20];
inline void add(int u,int v,int w){
	to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;val[cnt]=w;
}
void insert(int x,int &y,int l,int r,int v,int c){
	y=++tot;
	if(l==r){t[y] = t[x] + c; sum[y] = sum[x] + c * v; return ;}
	int mid=l+r>>1;
	if(v<=mid) rs[y]=rs[x],insert(ls[x],ls[y],l,mid,v,c);
	else ls[y]=ls[x],insert(rs[x],rs[y],mid+1,r,v,c);
	t[y]=t[ls[y]]+t[rs[y]];
	sum[y]=sum[ls[y]]+sum[rs[y]];
}
LL query(int x,int l,int r,int k){
	if(l==r) return 1ll*l*min(k,t[x]);
	int mid=l+r>>1;
	if(k<=t[ls[x]])	return query(ls[x],l,mid,k);
	else return query(rs[x],mid+1,r,k-t[ls[x]]) + sum[ls[x]];
}
int qx(int x,int l,int r,int k){
	if(l==r) return l;
	int mid=l+r>>1;
	if(k<=t[ls[x]]) return qx(ls[x],l,mid,k);
	else return qx(rs[x],mid+1,r,k-t[ls[x]]);
}
int main(){
	scanf("%d%d",&n,&m);
	int i,x,y,z,j,w;
	for(i=1;i<=n;i++){
		scanf("%d%d%d",&x,&y,&z);
		add(x,z,1);add(y+1,z,-1);
	}
	for(i=1;i<=m;i++){
		int f=0;
		for(j=head[i];j;j=nxt[j]){
			if(!f){insert(root[i-1],root[i],1,maxn,to[j],val[j]);f=1;}
			else insert(root[i],root[i],1,maxn,to[j],val[j]);
		}
		if(!f)	root[i]=root[i-1];
	}
	LL ans=1;
	for(i=1;i<=m;i++){
		scanf("%d%d%d%d",&x,&y,&z,&w);
		y=(y*ans+z)%w+1;
		ans=query(root[x],1,maxn,y);
		printf("%lld
",ans);
	}
}
原文地址:https://www.cnblogs.com/suika/p/8594690.html