[BZOJ3110][Zjoi2013]K大数查询(主席数套线段树 )

题目描述

传送门

题解

外层权值线段树,权值线段树的每一个位置都是一棵线段树,线段树用动态开点。
注意pushdown或者查询的时候还有可能要继续开点。
注意最顶端的点的权有可能是炸了int了,因为有可能加入了50000^2个点。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long

const int max_n=1e6+5;
const int max_tree=2e7+5;

int n,m,opt,cnt,N,t,ans,sz;
int A[max_n],B[max_n],C[max_n],number[max_n],hash[max_n];bool flag[max_n];
int root[max_n],ls[max_tree],rs[max_tree];
LL sum[max_tree],delta[max_tree];

inline int find(int x){
    int l=1,r=N,mid;
    while (l<=r){
        mid=(l+r)>>1;
        if (x>hash[mid]) l=mid+1;
        else r=mid-1;
    }
    return l;
}
inline void modify(int now){
    sum[now]=sum[ls[now]]+sum[rs[now]];
}
inline void pushdown(int now,int l,int r,int mid){
    if (delta[now]){
        if (!ls[now]) ls[now]=++sz;
        sum[ls[now]]+=delta[now]*(mid-l+1);
        delta[ls[now]]+=delta[now];
        if (!rs[now]) rs[now]=++sz;
        sum[rs[now]]+=delta[now]*(r-mid);
        delta[rs[now]]+=delta[now];
        delta[now]=0;
    }
}
inline void update(int last,int &now,int l,int r,int lrange,int rrange,LL v){
    int mid=(l+r)>>1;
    if (!now) now=++sz; ls[now]=ls[last]; rs[now]=rs[last];
    if (lrange<=l&&r<=rrange){
        sum[now]+=v*(r-l+1);
        delta[now]+=v;
        return;
    }
    pushdown(now,l,r,mid);
    if (lrange<=mid) update(ls[last],ls[now],l,mid,lrange,rrange,v);
    if (mid+1<=rrange) update(rs[last],rs[now],mid+1,r,lrange,rrange,v);
    modify(now);
}
inline LL query(int now,int l,int r,int lrange,int rrange){
    int mid=(l+r)>>1;LL ans=0;
    if (lrange<=l&&r<=rrange) return sum[now];
    pushdown(now,l,r,mid);
    if (lrange<=mid) ans+=query(ls[now],l,mid,lrange,rrange);
    if (mid+1<=rrange) ans+=query(rs[now],mid+1,r,lrange,rrange);
    return ans;
}
inline void v_update(int now,int l,int r,int x,int lrange,int rrange){
    int mid=(l+r)>>1;
    update(root[now],root[now],1,n,lrange,rrange,1);
    if (l==r) return;
    if (x<=mid) v_update(now<<1,l,mid,x,lrange,rrange);
    else v_update(now<<1|1,mid+1,r,x,lrange,rrange);
}
inline int v_query(int now,int l,int r,int lrange,int rrange,LL k){
    if (!now) return 0;
    int mid=(l+r)>>1;
    if (l==r) return l;
    LL t=query(root[now<<1],1,n,lrange,rrange);
    if (t>=k) return v_query(now<<1,l,mid,lrange,rrange,k);
    else return v_query(now<<1|1,mid+1,r,lrange,rrange,k-t);
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;++i){
        scanf("%d%d%d%d",&opt,&A[i],&B[i],&C[i]);
        if (opt==1) flag[i]=true,number[++cnt]=C[i];
    }
    sort(number+1,number+cnt+1);
    for (int i=1;i<=cnt;++i)
      if (number[i]!=number[i-1]) hash[++N]=number[i];

    for (int i=1;i<=m;++i)
      if (flag[i]){
        t=find(C[i]);
        v_update(1,1,N,t,A[i],B[i]);
      }
      else{
        LL tot=query(root[1],1,n,A[i],B[i]);
        LL t=(LL)tot-(LL)C[i]+1;
        ans=v_query(1,1,N,A[i],B[i],t);
        printf("%d\n",hash[ans]);
      }
}

总结

这种动态开点点数和空间的计算十分蛋疼。

你可能感兴趣的:(线段树,bzoj,主席树)