首页 > 代码库 > BZOJ 3196 Tyvj 1730 二逼平衡树 ——树状数组套主席树
BZOJ 3196 Tyvj 1730 二逼平衡树 ——树状数组套主席树
【题目分析】
听说是树套树。(雾)
怒写树状数组套主席树,然后就Rank1了。23333
单点修改,区间查询+k大数查询=树状数组套主席树。
【代码】
#include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <set> #include <map> #include <string> #include <algorithm> #include <vector> #include <iostream> #include <queue> using namespace std; #define maxn 50005 #define mlog 20 int read() { int x=0,f=1; char ch=getchar(); while (ch<‘0‘||ch>‘9‘) {if (ch==‘-‘) f=-1; ch=getchar();} while (ch>=‘0‘&&ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();} return x*f; } int n,m; int rt[maxn<<1],a[maxn],b[maxn<<1],cnt=0,tot=0,sum,rnk,TMP; int ls[maxn<<6],rs[maxn<<6],siz[maxn<<6]; int L[maxn],R[maxn],opt[maxn],x[maxn],y[maxn],z[maxn]; void ins(int o1,int & o2,int l,int r,int x,int f) { // printf("ins %d %d %d %d %d %d\n",o1,o2,l,r,x,f); o2=++tot; siz[o2]=siz[o1]+f; if (l==r) return ; int mid=(l+r)/2; if (x<=mid) rs[o2]=rs[o1],ins(ls[o1],ls[o2],l,mid,x,f); else ls[o2]=ls[o1],ins(rs[o1],rs[o2],mid+1,r,x,f); return ; } void ready(int l,int r) { // cout<<"ready for "<<l<<" "<<r<<endl; l--; L[0]=R[0]=0; for (int j=l;j;j-=j&(-j)) L[++L[0]]=rt[j]; for (int j=r;j;j-=j&(-j)) R[++R[0]]=rt[j]; // cout<<"L "; for (int j=1;j<=L[0];++j) cout<<L[j]<<" "; cout<<endl; // cout<<"R "; for (int j=1;j<=R[0];++j) cout<<R[j]<<" "; cout<<endl; // cout<<"L "; for (int j=1;j<=L[0];++j) cout<<siz[L[j]]<<" "; cout<<endl; // cout<<"R "; for (int j=1;j<=R[0];++j) cout<<siz[R[j]]<<" "; cout<<endl; } void cholef() { // cout<<"choose left"<<endl; for (int i=1;i<=L[0];++i) L[i]=ls[L[i]]; for (int i=1;i<=R[0];++i) R[i]=ls[R[i]]; } void chorig() { // cout<<"choose right"<<endl; for (int i=1;i<=L[0];++i) L[i]=rs[L[i]]; for (int i=1;i<=R[0];++i) R[i]=rs[R[i]]; } int taksum() { int ret=0; for (int i=1;i<=L[0];++i) ret-=siz[ls[L[i]]]; for (int i=1;i<=R[0];++i) ret+=siz[ls[R[i]]]; return ret; } int qrnk(int l,int r,int x) { // printf("qrnk %d %d %d\n",l,r,x); if (l==r) return 1; int mid=(l+r)/2,tmp=taksum(); // cout<<"take sum "<<tmp<<endl; if (x<=mid) return cholef(),qrnk(l,mid,x); else return chorig(),tmp+qrnk(mid+1,r,x); } int qnum(int l,int r,int x) { if (l==r) return l; int mid=(l+r)/2,tmp=taksum(); if (x<=tmp) return cholef(),qnum(l,mid,x); else return chorig(),qnum(mid+1,r,x-tmp); } int main() { n=read();m=read(); for (int i=1;i<=n;++i) scanf("%d",&a[i]),b[++cnt]=a[i]; for (int i=1;i<=m;++i) { scanf("%d%d%d",&opt[i],&x[i],&y[i]); if (opt[i]!=3) scanf("%d",&z[i]); if (opt[i]==3) b[++cnt]=y[i]; if (opt[i]==4||opt[i]==5) b[++cnt]=z[i]; } sort(b+1,b+cnt+1); cnt=unique(b+1,b+cnt+1)-b-1; for (int i=1;i<=n;++i) a[i]=lower_bound(b+1,b+cnt+1,a[i])-b; for (int i=1;i<=m;++i) { if (opt[i]==1||opt[i]==4||opt[i]==5)z[i]=lower_bound(b+1,b+cnt+1,z[i])-b; if (opt[i]==3) y[i]=lower_bound(b+1,b+cnt+1,y[i])-b; } // cout<<endl<<endl; // for (int i=1;i<=cnt;++i) cout<<i<<" ";cout<<endl; // for (int i=1;i<=cnt;++i) cout<<b[i]<<" "; cout<<endl; // for (int i=1;i<=m;++i) cout<<opt[i]<<" "<<x[i]<<" "<<y[i]<<" "<<z[i]<<endl; // cout<<endl<<endl; for (int i=1;i<=n;++i) for (int j=i;j<=n;j+=j&(-j)) ins(rt[j],rt[j],1,cnt,a[i],1); for (int i=1;i<=m;++i) { switch(opt[i]) { case 1: ready(x[i],y[i]); printf("%d\n",qrnk(1,cnt,z[i])); break; case 3: for (int j=x[i];j<=n;j+=j&(-j)) { ins(rt[j],rt[j],1,cnt,a[x[i]],-1); ins(rt[j],rt[j],1,cnt,y[i],1); } a[x[i]]=y[i]; break; case 2: ready(x[i],y[i]); printf("%d\n",b[qnum(1,cnt,z[i])]); break; case 4: ready(x[i],y[i]); rnk=qrnk(1,cnt,z[i]); // printf("rnk is %d\n",rnk); ready(x[i],y[i]); printf("%d\n",b[qnum(1,cnt,rnk-1)]); break; case 5: ready(x[i],y[i]); TMP=0; TMP-=qrnk(1,cnt,z[i]); ready(x[i],y[i]); TMP+=qrnk(1,cnt,z[i]+1); // printf("it s have %d\n",TMP); ready(x[i],y[i]); rnk=qrnk(1,cnt,z[i]); ready(x[i],y[i]); printf("%d\n",b[qnum(1,cnt,rnk+TMP)]); } } }
BZOJ 3196 Tyvj 1730 二逼平衡树 ——树状数组套主席树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。