bzoj 3932(主席树)

3932: [CQOI2015]任务查询系统

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 1452   Solved: 529
[ 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的一个排列

解题思路：模板题，但是wa了无数次

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define ll long long 
using namespace std;
int n,m,len,gen,cnt,tail,tail1;
ll sum[18700000],ans,Pre;
int l[18700000],r[18700000],val[18700000],root[110000];
int s[110000],e[110000],p[110000];
int to[210000],next[210000],h[210000];
int go[210000],con[210000],hg[210000],poo[110000];
int x[210000],y[210000];
struct ss
 {
    int zhi,dui;
 }q[110000];
 
inline ll read()
{
    char y; ll x=0,f=1; y=getchar();
    while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
    while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
    return x*f;
}
 
void insert(int x,int y)
 {
    ++len; to[len]=y; next[len]=h[x]; h[x]=len;
 }
  
void insert1(int x,int y)
 {
    ++gen; go[gen]=y; con[gen]=hg[x]; hg[x]=gen; 
 }
 
 
void build(int rot,int &now,int dui,int lo,int ro,int xl,int xr,int yl,int yr)
 {
    if (now==0) now=++cnt,sum[now]=sum[dui],val[now]=val[dui];
    for (int i=xl;i<=xr;++i) ++val[now],sum[now]+=poo[x[i]];
    for (int i=yl;i<=yr;++i) --val[now],sum[now]-=poo[y[i]];
    if (lo==ro) return;
    int mid=(lo+ro)/2;
    if (x[xl]>mid && y[yl]>mid || x[xr]<=mid && y[yr]<=mid)
{
  if (x[xl]>mid && y[yl]>mid)
   {
     l[now]=l[dui]; 
     build(rot,r[now],r[dui],mid+1,ro,xl,xr,yl,yr);
}else
{
r[now]=r[dui];
build(rot,l[now],l[dui],lo,mid,xl,xr,yl,yr);
}
}else
{
   int u1=xl; int u2=yl;
while (x[u1]<=mid && u1<=xr)++u1; --u1;
while (y[u2]<=mid && u2<=yr)++u2; --u2;
build(rot,l[now],l[dui],lo,mid,xl,u1,yl,u2);
   ++u1; ++u2;
build(rot,r[now],r[dui],mid+1,ro,u1,xr,u2,yr);
}
 }
 
void query(int now,int size,int lo,int ro)
 {
    int mid=(lo+ro)/2;
    if (lo==ro)
     {
        ans+=(long long) (size)*poo[lo];
        return;
      }
    if (size==0) return;
    if (val[l[now]]<=size) 
     {
        ans+=sum[l[now]];
        query(r[now],size-val[l[now]],mid+1,ro);
      }else
       {
        query(l[now],size,lo,mid);
       }
 }
 
bool cmp(ss x,ss y)
 {
    return x.zhi<y.zhi;
 }
 
int main()
{
freopen("main.in","r",stdin);
freopen("main.out","w",stdout);
    cnt=0;
    m=read(); n=read(); 
    for (int i=1;i<=m;++i)
     {
        s[i]=read(); e[i]=read(); p[i]=read();
        q[i].zhi=p[i]; q[i].dui=i; 
     }
    sort(q+1,q+m+1,cmp);
    int sug=0; 
    for (int i=1;i<=m;++i)
     {
        if (q[i].zhi!=q[i-1].zhi) ++sug; 
        poo[sug]=p[q[i].dui]; p[q[i].dui]=sug;
     }
    for (int i=1;i<=m;++i)
    {
        insert(s[i],p[i]); insert1(e[i]+1,p[i]);
    }
    for (int i=1;i<=n;++i)
     {
       int u=h[i]; tail=0; tail1=0;
       while (u!=0)
        {
          ++tail; x[tail]=to[u];
          u=next[u];
        }
       u=hg[i];
       while (u!=0)
        {
          ++tail1; y[tail1]=go[u]; 
          u=con[u]; 
        }
       sort(x+1,x+tail+1); sort(y+1,y+tail1+1);
       build(i,root[i],root[i-1],1,n,1,tail,1,tail1);
     }
    Pre=1;
    for (int i=1;i<=n;++i)
     {
        int xi,ai,bi,ci;
        xi=read(); ai=read(); bi=read(); ci=read();
        long long kg=1+(ai*Pre+bi)%ci;
        if (val[root[xi]]<kg) 
         {
            printf("%lld\n",sum[root[xi]]);
            Pre=sum[root[xi]];
          }else
            {
              ans=0;
              query(root[xi],kg,1,m);
              printf("%lld\n",ans);
              Pre=ans;  
            }
     }
}