BZOJ 3932: [CQOI2015]任务查询系统 【主席树】

主席树板题。
把任务差分之后插入，单点查询就好了。
注意空间要开100000*50，因为插入分了两部分（*40会WA掉。。）。
注意找前K小的时候要和根节点的siz取min。
还有查询的时候l==r是返回l*x而不是sum[now]
Code：

#include<cstdio>
#include<cctype>
#include<vector>
#include<algorithm>
#define maxn 100005
#define maxp maxn*50
#define LL long long
const int N = 10000000;
using namespace std;
inline void read(int &a){
    char c;bool f=0;
	while(!isdigit(c=getchar())) if(c=='-') f=1;
    for(a=c-'0';isdigit(c=getchar());a=a*10+c-'0');
	if(f) a=-a;
}
int n,m;
int rt[maxn],lc[maxp],rc[maxp],siz[maxp],tot;
LL s[maxp],ans=1;
struct node{
	int x,v,op;
	bool operator < (const node &p)const{return x<p.x;}
}a[maxn*2];
void insert(int &now,int l,int r,int x,int d){
	s[++tot]=s[now]+d*x,siz[tot]=siz[now]+d,lc[tot]=lc[now],rc[tot]=rc[now];
	now=tot;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(x<=mid) insert(lc[now],l,mid,x,d);
	else insert(rc[now],mid+1,r,x,d);
}
LL query(int now,int l,int r,int x){
	if(l==r) return 1ll*l*x;
	int mid=(l+r)>>1;
	if(x<=siz[lc[now]]) return query(lc[now],l,mid,x);
	else return s[lc[now]]+query(rc[now],mid+1,r,x-siz[lc[now]]);
}
int main()
{
	int x,y,z,A,B,C;
	read(m),read(n);
	for(int i=1;i<=m;i++){
		read(x),read(y),read(z);
		a[i]=(node){x,z,1},a[i+m]=(node){y+1,z,-1};
	}
	sort(a+1,a+1+m*2);
	for(int i=1,j=1;i<=n;i++){
		rt[i]=rt[i-1];
		for(;j<=2*m&&a[j].x<=i;j++) insert(rt[i],1,1e7,a[j].v,a[j].op);
	}
	for(int i=1;i<=n;i++){
		read(x),read(A),read(B),read(C);
		y=1+(A*ans+B)%C;
		printf("%lld\n",ans=query(rt[x],1,1e7,min(y,siz[rt[x]])));
	}
}