bzoj 3932: [CQOI2015]任务查询系统

题目描述
最近实验室正在为其管理的超级计算机编制一套任务管理系统，而你被安排完成其中的查询部分。
超级计算机中的任务用三元组(Si,Ei,Pi)描述，(Si,Ei,Pi)表示任务从第Si秒开始，在第Ei秒后结束（第Si秒和Ei秒任务也在
运行），其优先级为Pi。同一时间可能有多个任务同时执行，它们的优先级可能相同，也可能不同。
调度系统会经常向查询系统询问，第Xi秒正在运行的任务中，优先级最小的Ki个任务（即将任务按照优先级从小到大
排序后取前Ki个）的优先级之和是多少。特别的，如果Ki大于第Xi秒正在运行的任务总数，则直接回答第Xi秒正在运行
的任务优先级之和。上述所有参数均为整数，时间的范围在1到n之间（包含1和n）。
输入格式
输入文件第一行包含两个空格分开的正整数m和n，分别表示任务总数和时间范围。
接下来m行，每行包含三个空格分开的正整数Si、Ei和Pi(Si≤Ei)，描述一个任务。
接下来n行，每行包含四个空格分开的整数Xi、Ai、Bi和Ci，描述一次查询。查询的参数Ki需要由公式 Ki=1 +
(Ai*Pre+Bi) mod Ci 计算得到。其中Pre表示上一次查询的结果，对于第一次查询，Pre=1。
输出格式
输出共n行，每行一个整数，表示查询结果。
输入样例
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
输出样例
281
1
样例解释
K1 = (1*1+3)%2+1 = 1
K2 = (1*2+3)%4+1 = 2
K3 = (2*8+4)%3+1 = 3
数据范围
对于50%的数据，Ai=0

对于100%的数据，1≤m,n,Si,Ei,Ci≤100000，0≤Ai,Bi≤100000，1≤Pi≤10000000，Xi为1到n的一个排列

区间k小？主席树！

先离散pi。然后以离散后的pi建立线段树，线段树记录值的个数和区间和。对于每个任务。si的位置+1，pi的位置-1，每次更新记录前缀和就好了

#include<cstdio>
#include<algorithm>
using namespace std;
struct tree
{
     int l,r;
     int ll,rr;
     int x;
	 long long s;
}tr[8000001];
int tot;
inline void build(int p,int l,int r)
{
     tr[p].l=l;
     tr[p].r=r;
     int pp=tot;
     if(l!=r)
     {
     	  int mid=(l+r)/2;
     	  tot++;
          tr[pp].ll=tot;
          build(tot,l,mid);
          tot++;
          tr[pp].rr=tot;
          build(tot,mid+1,r);
     }
}
inline int inc(int p,int l,int r,int s,int x)
{
	 if(p==0)
	      return 0;
     if(l<=tr[p].l&&tr[p].r<=r)
     {
          tot++;
          tr[tot].l=l;
          tr[tot].r=r;
          tr[tot].s=tr[p].s+s*x;
          tr[tot].x=tr[p].x+x;
          return tot;
     }
     else
     {
     	  tot++;
     	  int pp=tot;
     	  tr[pp].l=tr[p].l;
     	  tr[pp].r=tr[p].r;
          int mid=(tr[p].l+tr[p].r)/2;
          if(l<=mid)
          {
               tr[pp].ll=inc(tr[p].ll,l,r,s,x);
               tr[pp].rr=tr[p].rr;
          }
          if(r>mid)
          {
               tr[pp].ll=tr[p].ll;
               tr[pp].rr=inc(tr[p].rr,l,r,s,x);
          }
          tr[pp].s=tr[tr[pp].ll].s+tr[tr[pp].rr].s;
          tr[pp].x=tr[tr[pp].ll].x+tr[tr[pp].rr].x;
          return pp;
     }
}
long long f[100001],fx[100001];
inline long long ask(int p,int k)
{
	 if(tr[p].l==tr[p].r)
	      return f[tr[p].l]*k;
	 if(tr[p].x<=k)
	      return tr[p].s;
     if(tr[tr[p].ll].x<k)
          return tr[tr[p].ll].s+ask(tr[p].rr,k-tr[tr[p].ll].x);
     return ask(tr[p].ll,k);
}
struct message
{
     long long s,e,p;
     long long x;
}a[100001];
int pl[100001];
struct change
{
     long long x,p,k,t;
     long long d;
}c[300001];
inline bool cmp1(message x,message y)
{
     if(x.p<y.p)
          return true;
     return false;
}
inline bool cmp2(change x,change y)
{
     if(x.x<y.x)
          return true;
     return false;
}
int main()
{
//	 freopen("query.in","r",stdin);
//	 freopen("query.out","w",stdout);
     int m,n;
     scanf("%d%d",&m,&n);
     int i;
     int sx=0;
     for(i=1;i<=m;i++)
     {
          scanf("%lld%lld%lld",&a[i].s,&a[i].e,&a[i].p);
		  a[i].x=i;
     }
     sort(a+1,a+1+m,cmp1);
     int p=0;
     p++;
     f[p]=a[1].p;
     fx[a[1].x]=p;
     for(i=2;i<=m;i++)
     {
          if(a[i].p!=a[i-1].p)
          {
               p++;
               f[p]=a[i].p;
          }
          fx[a[i].x]=p;
     }
     for(i=1;i<=m;i++)
     {
          sx++;
          c[sx].x=a[i].s;
          c[sx].p=a[i].p;
          c[sx].k=fx[a[i].x];
          c[sx].d=1;
		  sx++;
          c[sx].x=a[i].e+1;
          c[sx].p=a[i].p;
          c[sx].k=fx[a[i].x];
          c[sx].d=-1;
     }
     sort(c+1,c+1+sx,cmp2);
     tot++;
     pl[0]=tot;
     build(1,1,p);
     int lar=pl[0];
     for(i=1;i<=sx;i++)
     {
          int tp=tot+1;
          inc(lar,c[i].k,c[i].k,c[i].p,c[i].d);
          pl[c[i].x]=tp;
          lar=tp;
     }
     for(i=1;i<=c[sx].x;i++)
          if(pl[i]==0)
               pl[i]=pl[i-1];
     long long xi,ai,bi,ci,pre=1,ki;
     for(i=1;i<=n;i++)
     {
          scanf("%lld%lld%lld%lld",&xi,&ai,&bi,&ci);
          ki=(ai*pre+bi)%ci+1;
          pre=ask(pl[xi],ki);
          printf("%lld\n",pre);
     }
     return 0;
}