bzoj3110[Zjoi2013]K大数查询 主席树套线段树

xiaoxiao2021-02-28  54

题意很简单,就是带修改的K大数查询。 查询自不用说,怎么修改? 由于数字都在1-n范围内,所以他的权值就是他的排名,直接插入,然后建立区间线段树,在线段树内找到修改的范围,然后用主席树区间修改。 因为多打两个else而调了一上午的sb。。。 权值线段树本来也可以,但是数据加强以后被卡了。

#include<cstdio> #include<cstring> #include<algorithm> #define fo(i,a,b) for(int i=a;i<=b;i++) #define fd(i,a,b) for(int i=a;i>=b;i--) using namespace std; const int maxn=1e6+5; const int M=2e7+5; typedef long long ll; int a[maxn],b[maxn],c[maxn]; int N; bool flag[maxn]; int hash[maxn]; int num[maxn],cnt,op; int n,m,sz,t,ans,root[maxn],ls[M],rs[M]; ll sum[M],lazy[M]; inline void pushdown(int x,int l,int r,int mid) { if (!lazy[x])return; if (!ls[x])ls[x]=++sz; sum[ls[x]]+=(mid-l+1)*lazy[x]; lazy[ls[x]]+=lazy[x]; if (!rs[x])rs[x]=++sz; sum[rs[x]]+=(r-mid)*lazy[x]; lazy[rs[x]]+=lazy[x]; lazy[x]=0; } inline void update(int last,int &x,int l,int r,int a,int b,ll v) { if (!x)x=++sz; int mid=(l+r)>>1; ls[x]=ls[last],rs[x]=rs[last]; if (l>=a&&r<=b) { sum[x]+=v*(r-l+1); lazy[x]+=v; return; }pushdown(x,l,r,mid); if (a<=mid)update(ls[last],ls[x],l,mid,a,b,v); if (mid+1<=b)update(rs[last],rs[x],mid+1,r,a,b,v); sum[x]=sum[ls[x]]+sum[rs[x]]; } inline ll query(int x,int l,int r,int a,int b) { ll ans=0; int mid=(l+r)>>1; if (a<=l&&r<=b)return sum[x]; pushdown(x,l,r,mid); if (a<=mid)ans+=query(ls[x],l,mid,a,b); if (mid+1<=b)ans+=query(rs[x],mid+1,r,a,b); return ans; } inline void v_update(int x,int l,int r,int v,int a,int b) { int mid=(l+r)>>1; update(root[x],root[x],1,n,a,b,1); if (l==r)return; if (v<=mid)v_update(x<<1,l,mid,v,a,b); else v_update(x<<1|1,mid+1,r,v,a,b); } inline int v_query(int x,int l,int r,int a,int b,ll k) { if (!x)return 0; int mid=(l+r)>>1; if (l==r)return l; ll t=query(root[x<<1],1,n,a,b); if (t>=k) return v_query(x<<1,l,mid,a,b,k); else return v_query(x<<1|1,mid+1,r,a,b,k-t); } 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; } int main() { scanf("%d%d",&n,&m); fo(i,1,m) { scanf("%d%d%d%d",&op,&a[i],&b[i],&c[i]); if (op==1) { flag[i]=true; num[++cnt]=c[i]; } } sort(num+1,num+1+cnt); fo(i,1,cnt) { if (num[i]!=num[i-1])hash[++N]=num[i]; } fo(i,1,m) { 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]); } } }
转载请注明原文地址: https://www.6miu.com/read-74188.html

最新回复(0)