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

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

题解：本题是主席树的一个简化版吧，我们还是先离散化，然后按时间排序，每有一个任务开始或结束就新建一棵线段树，然后再查询时二分查找时间，然后再线段树上求前k个之和就行了，并且本题的主席树是不用相减的（这还叫主席树吗）

注意一下long long，注意二分边界不要搞错（尤其是用upper_bound的童鞋们），还要注意用离散化后的数来求出答案，还有第47行有点坑人

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll maxn=200010;
ll n,m,tot,nm,ans;
struct task
{
    ll pt,pv,pk;
}p[maxn];
ll ls[40*maxn],rs[40*maxn],sum[40*maxn],siz[40*maxn];
ll rp[maxn>>1],root[maxn];
bool cmp1(task a,task b)
{
    return a.pv<b.pv;
}
bool cmp2(task a,task b)
{
    return a.pt<b.pt;
}
ll readin()
{
    ll ret=0,f=1;   char gc=getchar();
    while(gc<'0'||gc>'9') {if(gc=='-')f=-f;gc=getchar();}
    while(gc>='0'&&gc<='9')   ret=ret*10+gc-'0',gc=getchar();
    return ret*f;
}
void insert(ll x,ll &y,ll l,ll r,ll pos,ll val)
{
    y=++tot;
    if(l==r)
    {
        siz[y]=siz[x]+val;
        sum[y]=sum[x]+rp[l]*val;
        return ;
    }
    ll mid=l+r>>1;
    if(pos<=mid) rs[y]=rs[x],insert(ls[x],ls[y],l,mid,pos,val);
    else    ls[y]=ls[x],insert(rs[x],rs[y],mid+1,r,pos,val);
    siz[y]=siz[ls[y]]+siz[rs[y]];
    sum[y]=sum[ls[y]]+sum[rs[y]];
}
ll query(ll x,ll l,ll r,ll pos)
{
    if(l==r)    return rp[l]*pos;
    ll mid=l+r>>1;
    if(siz[ls[x]]>=pos)  return query(ls[x],l,mid,pos);
    else    return sum[ls[x]]+query(rs[x],mid+1,r,pos-siz[ls[x]]);
}
int main()
{
    n=readin(),m=readin();
    ll i,a,b,c,d,e;
    for(i=1;i<=n;i++)
        p[i*2-1].pt=readin(),p[i*2].pt=readin()+1,p[i*2-1].pv=p[i*2].pv=readin(),
        p[i*2-1].pk=1,p[i*2].pk=-1;
    sort(p+1,p+2*n+1,cmp1);
    for(i=1;i<=2*n;i++)
    {
        if(rp[nm]<p[i].pv)   rp[++nm]=p[i].pv;
        p[i].pv=nm;
    }
    sort(p+1,p+2*n+1,cmp2);
    for(i=1;i<=2*n;i++)
        insert(root[i-1],root[i],1,nm,p[i].pv,p[i].pk);
    ans=1;
    for(i=1;i<=m;i++)
    {
        d=readin(),a=readin(),b=readin(),c=readin();
        e=1+(a*ans+b)%c;
        ll l=1,r=2*n+1,mid;
        while(l<r)
        {
            mid=l+r>>1;
            if(p[mid].pt<=d) l=mid+1;
            else    r=mid;
        }
        l--;
        if(siz[root[l]]<=e)  ans=sum[root[l]];
        else    ans=query(root[l],1,nm,e);
        printf("%lld\n",ans);
    }
    return 0;
}