【CQOI2015】任务查询系统（主席树+差分）

这是一道好题 可以更深的理解主席树

最初的想法是 一开始 一边加入任务 一边维护时间轴

换句话说 对于每个时间点 我们都想维护一颗权值线段树（这里的权值代表着优先级Pi）相当于对于一个任务 我们要维护Ei-Si棵

然而显然MLE 考虑降维

联想到主席树建树方式是通过前缀和来的

既然是求前缀和 这道题的每个任务相当于区间修改 会不会与差分数组有关呢？

答案是正确的

所以一开始我们想维护的Ei-Si棵权值线段树 变成只维护Ei,Si+1这两棵就行了 Ei进行+1，Si+1进行-1

因此我们按顺序 rebuild这些权值线段树 把他们合并成主席树

然后就可以query了

#include<bits/stdc++.h>
#define int long long
#define N 100005
using namespace std;
int m,n,a[N],num,rt[N*100],tot,lson[N*100],rson[N*100],cnt[N*100],sum[N*100];
struct Task
{
    int from,to,val;
}task[N];
inline void pushup(int now)
{
    sum[now]=sum[lson[now]]+sum[rson[now]];
    cnt[now]=cnt[lson[now]]+cnt[rson[now]];
}
inline void build(int &now,int l,int r,int pos,int del)
{
    if(!now) now=++tot;
    if(l==r)
    {
        cnt[now]+=del;
        sum[now]=cnt[now]*a[l];
        return;
    }
    int m=(l+r)>>1;
    if(pos<=m)  build(lson[now],l,m,pos,del);
    else build(rson[now],m+1,r,pos,del);
    pushup(now);
}
inline int rebuild(int x,int y)
{
    if(!x||!y)  return x+y;
    int now=++tot;
    sum[now]=sum[x]+sum[y];
    cnt[now]=cnt[x]+cnt[y];
    lson[now]=rebuild(lson[x],lson[y]);
    rson[now]=rebuild(rson[x],rson[y]);
    return now;
}
inline int query(int now,int l,int r,int k)
{
    if(l==r)    return a[l]*min(k,cnt[now]);
    int m=(l+r)>>1;
    if(cnt[lson[now]]>=k)   return query(lson[now],l,m,k);
    else return sum[lson[now]]+query(rson[now],m+1,r,k-cnt[lson[now]]);
}
main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);cout.tie(NULL);
    cin>>m>>n;
    for(int i=1;i<=m;i++)
    {
        cin>>task[i].from>>task[i].to>>task[i].val;
        a[i]=task[i].val;
    }
    sort(a+1,a+1+m);
    ++n;
    num=unique(a+1,a+1+m)-a-1;
    for(int i=1;i<=m;i++)
    {
        task[i].val=lower_bound(a+1,a+1+num,task[i].val)-a;
        build(rt[task[i].from],1,num,task[i].val,1);
        build(rt[task[i].to+1],1,num,task[i].val,-1);
    }
    for(int i=1;i<=n;i++)
    {
        rt[i]=rebuild(rt[i-1],rt[i]);
    }
    int pre=1;
    for(int i=1;i<n;i++)
    {
        int x,a,b,c,k;
        cin>>x>>a>>b>>c;
        k=1+(a*pre+b)%c;
        pre=query(rt[x],1,num,k);
        cout<<pre<<'\n';
    }
    return 0;
}