BZOJ 3932: [CQOI2015]任务查询系统 [主席树]

题意

给出$n$个由三元组$(s_i,e_i,p_i)$表示的区间,$s_i,e_i$表示头尾(包括头尾),$p_i$表示这个区间的权重

给出$m$个询问,询问包含位置$x$的按权重排序前$k$小的区间的权重和

输入先给出所有三元组,询问强制在线

题解

主席树的权值线段树按照$p_i$来搞,因为值域比较大,要离散一下

所有三元组先读入,拆成两个端点

每个端点维护两个权值,$sum1$表示权重和,$sum2$表示区间个数

$s$和$e+1$,$s$上为正的,$e+1$上为负的,相当于差分

然后把每个端点按时间轴插入主席树

注意下空间

#include<cstdio>
#include<algorithm>
#define N 100005
#define M N*40
#define LL long long
using namespace std;

int n,m,sz,ls[M],rs[M],P[N],pos[N],id[N],root[N];
LL tmp,sum1[M],sum2[M];
struct node{
    int tim,pos,v1,v2;
}a[N<<1];

inline int read(){
    int a=0;char f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}
    return a*f;
}

inline bool cmp(int x,int y){return P[x]<P[y];}

inline bool cmp1(node x,node y){return x.tim<y.tim;}

void insert(int &x,int l,int r,int lst,int p,int v1,int v2){
    x=++sz;
    ls[x]=ls[lst],rs[x]=rs[lst];
    sum1[x]=sum1[lst]+v1,sum2[x]=sum2[lst]+v2;
    if(l==r) return;
    int mid=l+r>>1;
    if(p<=mid) insert(ls[x],l,mid,ls[lst],p,v1,v2);
    else insert(rs[x],mid+1,r,rs[lst],p,v1,v2);
    return;
}

LL query(int x,int l,int r,int k){
    if(l==r) return sum1[x]+tmp;
    int mid=l+r>>1,cnt=sum2[ls[x]];
    if(cnt>=k) return query(ls[x],l,mid,k);
    else{
        tmp+=sum1[ls[x]];
        return query(rs[x],mid+1,r,k-cnt);
    }
}

int main(){
    freopen("testdata.in","r",stdin);
//  freopen("testdata.out","w",stdout);
    n=read(),m=read();
    int i,j;
    for(i=1,sz=0;i<=n;++i){
        int s=read(),t=read()+1;P[i]=read();
        a[++sz].tim=s,a[sz].v1=P[i],a[sz].v2=1;
        a[++sz].tim=t,a[sz].v1=-P[i],a[sz].v2=-1;
        id[i]=i;
    }
    sort(id+1,id+n+1,cmp);
    for(i=1;i<=n;++i) pos[id[i]]=i;
    for(i=1,sz=0;i<=n;++i) a[++sz].pos=pos[i],a[++sz].pos=pos[i];
    sort(a+1,a+sz+1,cmp1);
    sz=0;
    for(i=1,j=1;i<=m;++i){
        root[i]=root[i-1];
        while(a[j].tim==i){
            insert(root[i],1,n,root[i],a[j].pos,a[j].v1,a[j].v2);
            ++j;
        }
    }
    LL ans=1,A,B,C,K,x;
    while(m--){
        x=read(),A=read(),B=read(),C=read();
        K=1+(A*ans+B)%C;
        tmp=0;
        ans=query(root[x],1,n,K);
        printf("%lld\n",ans);
    }
    return 0;
}

为什么我总是跑得这么慢啊