BZOJ3932[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

样例输出

2
8
11

提示

样例解释

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

 

 

 

　　这道题要求强制在线，因此用主席树。记录每一时刻的事件，对于一个区间[l,r]在l时刻+1，在r+1时刻-1。然后按时间顺序插入主席树，维护区间权值和及区间数的个数，对于每个询问查询对应时刻线段树。注意pre初始值是1不是0！

#include<map>
#include<set>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define mid (L+R)/2
using namespace std;
int n,m;
int num;
int tot;
int len;
int x,y,z;
long long pre;
long long a,b,c;
int s[5000010];
int l[5000010];
int r[5000010];
int root[100010];
int cnt[5000010];
long long sum[5000010];
vector<int>p[100010],q[100010];
void build(int &x,int y)
{
    x=++tot;
    l[x]=l[y];
    r[x]=r[y];
    sum[x]=sum[y];
    cnt[x]=cnt[y];
    s[x]=num;
}
void updata(int L,int R,int x,int &y,int v,int c)
{
    if(s[x]!=num)
    {
        build(y,x);
    }
    cnt[y]+=c;
    sum[y]+=1ll*c*v;
    if(L==R)
    {
        return ;
    }
    if(v<=mid)
    {
        updata(L,mid,l[x],l[y],v,c);
    }
    else
    {
        updata(mid+1,R,r[x],r[y],v,c);
    }
}
long long query(int L,int R,int x,int k)
{
    if(L==R)
    {
        return 1ll*L*k;
    }
    if(cnt[l[x]]>=k)
    {
        return query(L,mid,l[x],k);
    }
    else
    {
        return query(mid+1,R,r[x],k-cnt[l[x]])+sum[l[x]];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int mn=1e9;
    int mx=-1e9;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        mn=min(mn,x);
        mx=max(mx,y);
        p[x].push_back(z);
        q[y].push_back(z);
    }
    for(int i=mn;i<=mx;i++)
    {
        build(root[i],root[i-1]);
        num++;
        len=p[i].size();
        for(int j=0;j<len;j++)
        {
            updata(1,1e7,root[i],root[i],p[i][j],1);
        }
        len=q[i-1].size();
        for(int j=0;j<len;j++)
        {
            updata(1,1e7,root[i],root[i],q[i-1][j],-1);
        }
    }
    pre=1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%lld%lld%lld",&x,&a,&b,&c);
        int k=(int) 1+(a*pre+b)%c;
        if(cnt[root[x]]<=k)
        {   
            pre=sum[root[x]];
            printf("%lld\n",pre);
        }
        else
        {
            pre=query(1,1e7,root[x],k);
            printf("%lld\n",pre);
        }
    }
}