洛谷 P3168 [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行，每行一个整数，表示查询结果。

输入输出样例

输入样例#1：
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
输出样例#1：
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+1$减去权值，然后第$i$个时刻的值，就是前$i$个的值前缀和，这个东西用主席树维护，第 k k <script type="math/tex" id="MathJax-Element-167">k</script>大和可以开权值线段树。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define LL long long

const LL maxn=2e5+7;
const LL maxp=1e7+7;

using namespace std; 

LL n,m,x,y,z,cnt,A,B,C,tot;
LL root[maxn];
LL l[maxn],r[maxn];

struct rec{
    LL x,z;
}q[maxn*2];

struct node{
    LL l,r;
    LL sum,data;
}t[5000007];

bool cmp(rec x,rec y)
{
    return x.x<y.x;
}

void ins(LL &p,LL q,LL l,LL r,LL k)
{
    if (!p) p=++tot;
    t[p].data=t[q].data;
    t[p].sum=t[q].sum;
    if (k>0) t[p].data++; 
        else t[p].data--;
    t[p].sum+=k;
    if (l==r) return;
    LL d=abs(k);
    LL mid=(l+r)/2;
    if (d<=mid) t[p].r=t[q].r,ins(t[p].l,t[q].l,l,mid,k);
           else t[p].l=t[q].l,ins(t[p].r,t[q].r,mid+1,r,k);
}

LL getsum(LL p,LL l,LL r,LL k)
{
    if (k>=t[p].data) return t[p].sum;
    if (l==r)
    {
        return t[p].sum/t[p].data*k;
    }
    LL mid=(l+r)/2;
    if (k<=t[t[p].l].data) return getsum(t[p].l,l,mid,k);
                      else return getsum(t[p].l,l,mid,t[t[p].l].data)+getsum(t[p].r,mid+1,r,k-t[t[p].l].data);
}

int main()
{
    scanf("%lld%lld",&n,&m);
    for (LL i=1;i<=n;i++)
    {
        scanf("%lld%lld%lld",&x,&y,&z);
        q[++cnt]=(rec){x,z};
        q[++cnt]=(rec){y+1,-z};
    }
    sort(q+1,q+cnt+1,cmp);
    LL j=1,k=1;                 
    for (LL i=1;i<=cnt;i++)
    {
        ins(root[i],root[i-1],1,maxp,q[i].z);
        while (j<=q[i].x)
        {
            l[j]=i;
            j++;
        }
        while (k<q[i].x)
        {
            r[k]=i-1;
            k++;
        }

    }       
    LL lastans=1;   
    for (LL i=1;i<=m;i++)
    {
        scanf("%lld%lld%lld%lld",&x,&A,&B,&C);
        LL k=1+(A*lastans+B)%C;
        lastans=getsum(root[r[x]],1,maxp,k);
        printf("%lld\n",lastans);
    }
}