BZOJ3932: [CQOI2015]任务查询系统 主席树

3932: [CQOI2015]任务查询系统

Time Limit: 20 Sec   Memory Limit: 512 MB

Submit: 1551   Solved: 572

[ Submit][ Status][ Discuss]

Description

最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。超级计算机中的

任务用三元组(Si,Ei,Pi)描述，(Si,Ei,Pi)表示任务从第Si秒开始，在第Ei秒后结束（第Si秒和Ei秒任务也在运行

），其优先级为Pi。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。调度系统会经常向

查询系统询问，第Xi秒正在运行的任务中，优先级最小的Ki个任务（即将任务按照优先级从小到大排序后取前Ki个

）的优先级之和是多少。特别的，如果Ki大于第Xi秒正在运行的任务总数，则直接回答第Xi秒正在运行的任务优先

级之和。上述所有参数均为整数，时间的范围在1到n之间（包含1和n）。

Input

输入文件第一行包含两个空格分开的正整数m和n，分别表示任务总数和时间范围。接下来m行，每行包含三个空格

分开的正整数Si、Ei和Pi(Si≤Ei)，描述一个任务。接下来n行，每行包含四个空格分开的整数Xi、Ai、Bi和Ci，

描述一次查询。查询的参数Ki需要由公式 Ki=1+(Ai*Pre+Bi) mod Ci计算得到。其中Pre表示上一次查询的结果，

对于第一次查询，Pre=1。

Output

输出共n行，每行一个整数，表示查询结果。

Sample Input

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

Sample Output

2
8
11

HINT

样例解释

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的一个排列

题目大意：

电脑有n个任务需要执行，任务i在si到ei时正在工作，优先级为p
给m个询问，每个询问给出一个时间点xi和一个数ki。
问在xi这个时间点时，所有正在工作的任务中优先级从小到大排列，前ki个的优先级之和是多少

题解：

看到了前k大，直接想到了主席树！

因为主席树中每一颗线段树都是由上一棵线段树转化而来。
于是！ 我们想到了差分！ 可以完美的和这个题目匹配；
把每一个任务的起始时间和终止时间记录并且差分一下（比如区间（1,2）优先级为6，就将差分数组中a[1]+=6,a[3]-=6），然后按照时间先后顺序排列，利用差分数组来做主席树就OK了。

用优先级大小做一个权值线段树，线段树上需要维护在这个区间上有多少个数（size），这些数的和（sum）是多少。
这样可以在第xi颗线段树上直接找前ki大的数，并将ki完全覆盖的区间的sum加起来就可以了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const int N=300050;
ll n,m,cnt,cn,lim;
ll root[N],w[N*35],son[N*35][2],sum[N*35],to[N];
struct pp {ll pos,v;}e[N*2];
bool cmp(pp u,pp v) {return u.pos<v.pos;}
void pushup(ll x)
{
	w[x]=w[son[x][0]]+w[son[x][1]];
	sum[x]=sum[son[x][0]]+sum[son[x][1]];
}
void addnew(ll x,ll &y,ll l,ll r,ll v)
{
	y=++cn;
	if(l==r)
	{
		if(v>0) w[y]=w[x]+1;
		else w[y]=w[x]-1;
		sum[y]=sum[x]+v;
		return;
	}
	son[y][0]=son[x][0];
	son[y][1]=son[x][1];
	ll mid=(l+r)>>1;
	if(abs(v)<=mid) addnew(son[x][0],son[y][0],l,mid,v);
	else addnew(son[x][1],son[y][1],mid+1,r,v);
	pushup(y);
}
ll check(ll R,ll k)
{
	ll x=root[R],l=1,r=lim,ans=0;
	if(w[x]<=k)
	{
		ans+=sum[x];
		return ans;
	}
	while(l<r)
	{
		ll mid=(l+r)>>1;
		if(w[son[x][0]]>=k)
		{
			r=mid;
			x=son[x][0];
		}
		else
		{
			l=mid+1;
			k-=w[son[x][0]];
			ans+=sum[son[x][0]];
			x=son[x][1];
		}
	}
	if(k) ans+=l*k;
	return ans;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	ll x,y,p,a,b,c;
	for(ll i=1;i<=n;i++)
	{
		scanf("%lld%lld%lld",&x,&y,&p);
		e[++cnt].pos=x,e[cnt].v=p;
		e[++cnt].pos=y+1,e[cnt].v=-p;
		lim=max(lim,p);
	}
	sort(e+1,e+cnt+1,cmp);
	ll now,pre=1;
	for(ll i=1;i<=cnt;i++) addnew(root[i-1],root[i],1,lim,e[i].v);
	for(ll i=cnt;i>=1;i--)
	if(e[i].pos!=e[i+1].pos) to[e[i].pos]=i;
	for(ll i=1;i<=n;i++)
	if(!to[i]) to[i]=to[i-1];
	for(ll i=1;i<=m;i++)
	{
		scanf("%lld%lld%lld%lld",&x,&a,&b,&c);
		a=(a*pre+b)%c+1;
		pre=check(to[x],a);
		printf("%lld\n",pre);
	}
}