1. 程式人生 > >bzoj 3196 Tyvj 1730 二逼平衡樹

bzoj 3196 Tyvj 1730 二逼平衡樹

很慢 怎麽 tar https void .com i++ \n node

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=3196

第一道的樹套樹。

線段樹套平衡樹。找了一個當模板。感覺還好啦。但跑得很慢。卡時過了。

1.好神奇呀,竟然都要加-INF和INF。別忘了求siz時-1;

2.去右兒子找的時候別忘了減去左兒子和自己的siz;

3.凡是有splay的地方就別忘了 & ;

4.想要 < 的就把它放在return y?y:cr裏,想要 <= 的也就把它放在那個地方,大約就是這樣就行;

5.線段樹的 tt 別和Splay的 tot 混用,不然線段樹節點的數組就得……

6.2e6是怎麽算的呀?1e6為什麽不行?

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=5e4+5,M=2e6+5,INF=1e9;
int n,m,t[N],tot,tt,fa[M],c[M][2],val[M],siz[M];//
struct Node{
    int l,r,ls,rs,rt;
}a[N<<1];
void updt(int cr)
{
    siz[cr]=siz[c[cr][0]]+siz[c[cr][1]]+1;
}
void rotate(int
x,int &rt) { int y=fa[x],z=fa[y],d=(x==c[y][1]); if(y==rt)rt=x; else c[z][y==c[z][1]]=x; fa[y]=x;fa[x]=z;fa[c[x][!d]]=y; c[y][d]=c[x][!d];c[x][!d]=y; updt(y);updt(x); } void splay(int x,int &rt) { while(x!=rt) { int y=fa[x],z=fa[y]; if(y!=rt) {
if((x==c[y][0])^(y==c[z][0]))rotate(x,rt); else rotate(y,rt); } rotate(x,rt); } } int find_k(int cr,int x) { if(siz[c[cr][0]]==x-1)return cr; if(siz[c[cr][0]]>=x)return find_k(c[cr][0],x); return find_k(c[cr][1],x-c[cr][0]-1); } int find_pre(int cr,int x) { if(!cr)return 0; if(val[cr]>=x)return find_pre(c[cr][0],x); int y=find_pre(c[cr][1],x); return y?y:cr; } void insert(int &rt,int x) { int y=find_pre(rt,x);splay(y,rt); y=find_k(c[rt][1],1);splay(y,c[rt][1]); fa[++tot]=y;c[y][0]=tot;siz[tot]=1;val[tot]=x; updt(y);updt(rt); } void del(int &rt,int x) { int y=find_pre(rt,x);splay(y,rt); y=find_k(c[rt][1],2);splay(y,c[rt][1]); fa[c[y][0]]=0;c[y][0]=0; updt(y);updt(rt); } void build(int l,int r,int cr) { a[cr].l=l;a[cr].r=r;a[cr].rt=++tot; fa[tot]=0;c[tot][1]=tot+1;c[tot][0]=0;val[tot]=-INF;siz[tot]=2; fa[++tot]=tot-1;c[tot][1]=c[tot][0]=0;val[tot]=INF;siz[tot]=1; for(int i=l;i<=r;i++)insert(a[cr].rt,t[i]); if(l==r)return; int mid=((l+r)>>1); a[cr].ls=++tt;build(l,mid,tt); a[cr].rs=++tt;build(mid+1,r,tt); } int find_Suc(int rt,int x) //Suc表示找的不是正常後繼(>=) { if(!rt)return 0; if(val[rt]<x)return find_Suc(c[rt][1],x); int y=find_Suc(c[rt][0],x); return y?y:rt; } int find_suc(int rt,int x) { if(!rt)return 0; if(val[rt]<=x)return find_suc(c[rt][1],x); int y=find_suc(c[rt][0],x); return y?y:rt; } int query_rnk(int &rt,int x) //& { int y=find_Suc(rt,x);splay(y,rt); return siz[c[rt][0]]-1; //-INF } int query_Rnk(int &rt,int x) { int y=find_suc(rt,x);splay(y,rt); return siz[c[rt][0]]-1; } int query_rnk(int L,int R,int cr,int x) { if(a[cr].l>=L&&a[cr].r<=R)return query_rnk(a[cr].rt,x); int mid=((a[cr].l+a[cr].r)>>1),ret=0; if(mid>=L)ret+=query_rnk(L,R,a[cr].ls,x); if(mid<R)ret+=query_rnk(L,R,a[cr].rs,x); return ret; } int query_Rnk(int L,int R,int cr,int x) { if(a[cr].l>=L&&a[cr].r<=R)return query_Rnk(a[cr].rt,x); int mid=((a[cr].l+a[cr].r)>>1),ret=0; if(mid>=L)ret+=query_Rnk(L,R,a[cr].ls,x); if(mid<R)ret+=query_Rnk(L,R,a[cr].rs,x); return ret; } void mdfy(int P,int cr,int x) { insert(a[cr].rt,x);del(a[cr].rt,t[P]); if(a[cr].l==a[cr].r)return; int mid=((a[cr].l+a[cr].r)>>1); if(P<=mid)mdfy(P,a[cr].ls,x); else mdfy(P,a[cr].rs,x); } int query_pre(int L,int R,int cr,int x) { if(a[cr].l>=L&&a[cr].r<=R)return val[find_pre(a[cr].rt,x)]; int ret=-INF,mid=((a[cr].l+a[cr].r)>>1); if(mid>=L)ret=max(ret,query_pre(L,R,a[cr].ls,x)); if(mid<R)ret=max(ret,query_pre(L,R,a[cr].rs,x)); return ret; } int query_suc(int L,int R,int cr,int x) { if(a[cr].l>=L&&a[cr].r<=R)return val[find_suc(a[cr].rt,x)]; int ret=INF,mid=((a[cr].l+a[cr].r)>>1); if(mid>=L)ret=min(ret,query_suc(L,R,a[cr].ls,x)); if(mid<R)ret=min(ret,query_suc(L,R,a[cr].rs,x)); return ret; } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++)scanf("%d",&t[i]); tt=1;build(1,n,1); int op,x,y,k; while(m--) { scanf("%d%d%d",&op,&x,&y);if(op!=3)scanf("%d",&k); if(op==1)printf("%d\n",query_rnk(x,y,1,k)+1); if(op==2){ int l=0,r=INF,ans=0; while(l<=r) { int mid=((l+r)>>1); if(query_Rnk(x,y,1,mid)>=k)ans=mid,r=mid-1;//不+1:包含了全部==自己的 else l=mid+1; } printf("%d\n",ans); } if(op==3)mdfy(x,1,y),t[x]=y; if(op==4)printf("%d\n",query_pre(x,y,1,k)); if(op==5)printf("%d\n",query_suc(x,y,1,k)); } return 0; }

bzoj 3196 Tyvj 1730 二逼平衡樹