Bzoj 3932: [CQOI2015]任务查询系统(主席树)

3932: [CQOI2015]任务查询系统
Time Limit: 20 Sec Memory Limit: 512 MB
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个优先级是谁&&优先级的价值和.
对于每个优先级建一棵权值线段树.
对于每个时间单独维护更新线段树.
把每个任务拆成两个时间.
做一个差分.
对于没有出现过的时间就直接复制之前的 
(这一点没想到然后看的题解orzorzorz).
按时间排序将每个点插入更新.
这题较坑的地方是query的时候
要返回tree[k].sum/tree[k].size*x.
因为我们只能查到第k个是谁.
但是里边有很多任务吖.
里边的优先级数都是一样的.
我们只能选择里边的k`个.

*/
#include<algorithm>
#include<iostream>
#include<cstdio>
#define LL long long
#define MAXN 100001
using namespace std;
int root[MAXN],n,m,cut,a[MAXN],tot;
LL ans=1;
struct node{int x,p,tag;}s[MAXN*2];
struct data{int lc,rc,size;LL sum;}tree[MAXN*40];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x*f;
}
bool cmp(const node &x,const node &y)
{
    return x.x<y.x;
}
int build(int l,int r)
{
    int k=++tot;
    if(l==r) return k;
    int mid=(l+r)>>1;
    tree[k].lc=build(l,mid);
    tree[k].rc=build(mid+1,r);
    return k;
}
void change(int &now,int last,int l,int r,int x,int tag)
{
    now=++tot;
    tree[now].lc=tree[last].lc,tree[now].rc=tree[last].rc;
    tree[now].sum=tree[last].sum+(LL)tag*a[x];
    tree[now].size=tree[last].size+tag;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(x<=mid) change(tree[now].lc,tree[now].lc,l,mid,x,tag); 
    else change(tree[now].rc,tree[now].rc,mid+1,r,x,tag);
    return ;
}
LL query(int k,int l,int r,int x)
{
    if(l==r) return tree[k].sum/tree[k].size*(LL)x;
    int p=tree[tree[k].lc].size;
    int mid=(l+r)>>1;
    if(p>=x) return query(tree[k].lc,l,mid,x);
    else return tree[tree[k].lc].sum+query(tree[k].rc,mid+1,r,x-p);
}
int main()
{
    freopen("cqoi15_query.in","r",stdin);
    freopen("cqoi15_query.out","w",stdout);
    int t,x,y,z;
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
        x=read(),y=read(),z=read();
        s[++cut].x=x,s[cut].p=z,s[cut].tag=1;
        s[++cut].x=y+1,s[cut].p=z,s[cut].tag=-1;
        a[i]=z;
    }
    sort(a+1,a+n+1);
    sort(s+1,s+cut+1,cmp);
    root[0]=build(1,m);
    int j=1;
    for(int i=1;i<=m;i++)
    {
        root[i]=root[i-1];
        while(j<=cut&&s[j].x==i)
        {
            int p=lower_bound(a+1,a+n+1,s[j].p)-a;
            change(root[i],root[i],1,m,p,s[j].tag);
            j++;
        }
    }
    for(int i=1;i<=m;i++)
    {
        t=read();x=read(),y=read(),z=read();
        LL k=1+((LL)x*ans+(LL)y)%z;
        if(k>=tree[root[t]].size) ans=tree[root[t]].sum;
        else ans=query(root[t],1,m,k);
        printf("%lld
",ans);
    }
    return 0;
}