BZOJ3932 CQOI2015 任务查询系统-可持久化线段树

BZOJ3932 CQOI2015 任务查询系统

可持久化线段树

每个时间点建立一棵权值线段树，保存数的个数与数的和。
我们发现相邻的时间点所对应的线段树用很多重复部分
于是我们把每个修改(Si,Ei,Pi)变成在Si处加入Pi，在1+Ei处减去Pi，时间点为Si+1~Ei的线段树直接由前一个时间点复制而来。回答询问时直接在第Xi棵线段树上查找即可。

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn = 100005;
const int INF  = 0x3f3f3f3f;
struct opt{
    int pos,key,fg;
}a[maxn<<1];
struct num{
    int pos,key;
}b[maxn];
int n,m,maxp=0;
bool cmp(opt x,opt y){
    return x.pos<y.pos;
}
bool cmpb(num x,num y){
    return x.key<y.key;
}
struct node{
    int siz;LL sum;
    node *L,*R;
}mempool[maxn*38];int tot=0;
node* new_node(const int c,const int d){
    node *t=&mempool[tot++];
    t->siz=c;t->sum=d;
    t->L=t->R=NULL;
    return t;
}
inline int siz(node* p){return p==NULL?0:p->siz;}
inline LL sum(node* p){return p==NULL?0:p->sum;}
node* add(node* p,int L,int R,int d,int k){
    int inc=k>0?1:-1;
    node* t=new_node( siz(p)+ inc , sum(p)+k);
    int mid=(L+R)>>1;
    if(L==R)return t;

    if(d<=mid){
        t->R=p->R;
        if(p->L==NULL)p->L=new_node(0,0);
        t->L=add(p->L,L,mid,d,k);
        t->L->siz=siz(p->L)+ inc ;
        t->L->sum=sum(p->L)+k;
    }else{
        t->L=p->L;
        if(p->R==NULL)p->R=new_node(0,0);
        t->R=add(p->R,mid+1,R,d,k);
        t->R->siz=siz(p->R)+ inc ;
        t->R->sum=sum(p->R)+k;
    }
    return t;
}
LL query(node *r,int L,int R,int k){
    if(r==NULL)return 0;
    if(k>=siz(r))return sum(r);
    int sz=siz(r);
    LL sm=sum(r);
    if(L==R)return 1ll*sm/sz*min(k,sz);
    int mid=(L+R)>>1;
    sz=siz(r->L);
    LL ret=0;
    ret+=query(r->L,L,mid,k);
    if(k>sz)ret+=query(r->R,mid+1,R,k-sz);
    return ret;
}
node* root[maxn];
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        int s,e,p,j=i+n;
        scanf("%d%d%d",&s,&e,&p);e++;
        a[i].pos=s; a[i].key=p; a[i].fg=p;
        a[j].pos=e; a[j].key=p; a[j].fg=-p;

        b[i].key=p;b[i].pos=i;
    }
    sort(b+1,b+1+n,cmpb);
    int T=0;
    int tmp=b[1].pos;
    a[tmp].key=a[tmp+n].key=++T;
    for(int i=2;i<=n;i++){
        tmp=b[i].pos;
        a[tmp].key=a[tmp+n].key=b[i].key==b[i-1].key?T:++T;
    }
    sort(a+1,a+1+n+n,cmp);

    root[0]=new_node(0,0);
    for(int i=1,t=1;i<=n<<1;i++){
        for(;t<a[i].pos;t++)root[t]=root[t-1];
        int d=a[i].key;
        root[t]=add(root[t-1],1,n,d,a[i].fg);
        i++;
        while(a[i].pos==a[i-1].pos){
            d=a[i].key;
            root[t]=add(root[t],1,n,d,a[i].fg);
            i++;
        }
        t++;i--;
    }

    LL pre=1;
    for(int i=1;i<=m;i++){
        int X,A,B,C;
        scanf("%d%d%d%d",&X,&A,&B,&C);
        int K=1+(A%C*pre%C+B%C)%C;
        LL ans=query(root[X],1,n,K);
        printf("%lld\n",ans);
        pre=ans;
    }
}