bzoj3932 [CQOI2015]任务查询系统

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的一个排列。
正解：主席树。
我真是太制杖了。。二分+主席树硬是把复杂度弄成了O(nlog^2n)。。
我是这么做的：按照优先级为前缀建立主席树。每次新建线段树就根据上一个优先级建立，区间修改，l-r之间维护一个优先级的和与任务次数。查询时先二分，找到当前点正好有k个任务的线段树，如果没有那就是最后一棵线段树。然后直接查询单点和就好了。
好像网上的做法是按照时间为前缀然后离散化优先级。。我真是太制杖了。。

//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <complex>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define N (100010)
#define il inline
#define RG register
#define ll long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)

using namespace std;

struct node{ ll l,r,p; }g
;

ll sum[50*N],res[50*N],lazy[50*N],ls[50*N],rs[50*N],rt
,n,m,sz,tot,ans=1;

il ll gi(){
RG ll x=0,q=1; RG char ch=getchar(); while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
if (ch=='-') q=-1,ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar(); return q*x;
}

il ll cmp(const node &a,const node &b){ return a.p<b.p; }

il void update(RG ll x,RG ll &y,RG ll l,RG ll r,RG ll xl,RG ll xr,RG ll v){
sum[y=++sz]=sum[x],res[y]=res[x],lazy[y]=lazy[x],ls[y]=ls[x],rs[y]=rs[x];
if (xl<=l && r<=xr){ sum[y]+=(r-l+1)*v,res[y]++,lazy[y]+=v; return; }
RG ll mid=(l+r)>>1;
if (xr<=mid) update(ls[x],ls[y],l,mid,xl,xr,v);
else if (xl>mid) update(rs[x],rs[y],mid+1,r,xl,xr,v);
else update(ls[x],ls[y],l,mid,xl,mid,v),update(rs[x],rs[y],mid+1,r,mid+1,xr,v);
sum[y]=sum[ls[y]]+sum[rs[y]]+(r-l+1)*lazy[y]; return;
}

il ll querytot(RG ll x,RG ll l,RG ll r,RG ll p,RG ll la){
if (l==r) return res[x]+la; RG ll mid=(l+r)>>1; la+=res[x];
if (p<=mid) return querytot(ls[x],l,mid,p,la);
else return querytot(rs[x],mid+1,r,p,la);
}

il ll querysum(RG ll x,RG ll l,RG ll r,RG ll p,RG ll la){
if (l==r) return sum[x]+la; RG ll mid=(l+r)>>1; la+=lazy[x];
if (p<=mid) return querysum(ls[x],l,mid,p,la);
else return querysum(rs[x],mid+1,r,p,la);
}

il ll find(RG ll x,RG ll siz){
RG ll l=1,r=n,mid,ans=n;
while (l<=r){
mid=(l+r)>>1;
if (querytot(rt[mid],1,m,x,0)>=siz) ans=mid,r=mid-1;
else l=mid+1;
}
return ans;
}

il void work(){
n=gi(),m=gi(); for (RG ll i=1;i<=n;++i) g[i].l=gi(),g[i].r=gi(),g[i].p=gi(); RG ll x,a,b,c,k;
sort(g+1,g+n+1,cmp); for (RG ll i=1;i<=n;++i) update(rt[i-1],rt[i],1,m,g[i].l,g[i].r,g[i].p);
for (RG ll i=1;i<=m;++i){
x=gi(),a=gi(),b=gi(),c=gi(),k=1+(a*ans+b)%c,tot=find(x,k);
ans=querysum(rt[tot],1,m,x,0); printf("%lld\n",ans);
}
return;
}

int main(){
File("system");
work();
return 0;
}