1. 程式人生 > >【Treap/非旋轉Treap】BZOJ3224 [Tyvj1728]普通平衡樹

【Treap/非旋轉Treap】BZOJ3224 [Tyvj1728]普通平衡樹

題面在這裡

平衡樹裸題……
寫了Treap/非旋轉Treap兩個版本。

【普通Treap】

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=100005,INF=0x3f3f3f3f;
int q;
struct node{
    node* s[2];
    int k,f,size,cnt;
    void maintain() {size=s[0]->size+s[1]->size+cnt;}
};
typedef node* P_node;
struct
treap{ node tree[maxn],nil; P_node null,len,root; treap(){ root=null=&nil; null->s[0]=null->s[1]=null; null->size=null->cnt=0; len=tree; } P_node newnode(int key){ len->k=key;len->f=rand(); len->s[0]=len->s[1
]=null; len->size=len->cnt=1; return len++; } void rot(P_node &x,int d){ P_node k=x->s[d^1];x->s[d^1]=k->s[d];k->s[d]=x; x->maintain();k->maintain();x=k; } void insert(P_node &x,int key){ if (x==null) x=newnode(key);else
if (x->k==key) x->cnt++;else{ int d=key>x->k; insert(x->s[d],key);if (x->s[d]->f > x->f) rot(x,d^1); } x->maintain(); } void erase(P_node &x,int key){ if (x->k==key){ if (x->cnt>1) x->cnt--;else if (x->s[0]==null) x=x->s[1];else if (x->s[1]==null) x=x->s[0];else{ int d=x->s[0]->f < x->s[1]->f; rot(x,d^1);erase(x->s[d^1],key); } }else erase(x->s[key>x->k],key); x->maintain(); } int kth(P_node &x,int k){ if (x==null||k<1||k>x->size) return 0; if (k<=x->s[0]->size) return kth(x->s[0],k);else if (k>x->s[0]->size+x->cnt) return kth(x->s[1],k-x->s[0]->size-x->cnt); return x->k; } int rank(P_node &x,int key){ if (x->k==key) return x->s[0]->size+1; if (key<x->k) return rank(x->s[0],key); return rank(x->s[1],key)+x->s[0]->size+x->cnt; } int pred(P_node &x,int key){ if (x==null) return -INF; if (key<=x->k) return pred(x->s[0],key); return max(x->k,pred(x->s[1],key)); } int succ(P_node &x,int key){ if (x==null) return INF; if (key>=x->k) return succ(x->s[1],key); return min(x->k,succ(x->s[0],key)); } void print(P_node &x){ if (x==null) return; print(x->s[0]); for (int i=1;i<=x->cnt;i++) printf("%d ",x->k); print(x->s[1]); } void Insert(int x) {insert(root,x);} void Erase(int x) {erase(root,x);} int Rank(int x) {return rank(root,x);} int Kth(int x) {return kth(root,x);} int Pred(int x) {return pred(root,x);} int Succ(int x) {return succ(root,x);} void Print() {print(root);printf("**\n");} }T; inline int red(){ int tot=0,f=1;char ch=getchar(); while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=getchar();} while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar(); return tot*f; } int main(){ q=red(); while (q--){ int c=red(),x=red(); if (c==1) T.Insert(x);else if (c==2) T.Erase(x);else if (c==3) printf("%d\n",T.Rank(x));else if (c==4) printf("%d\n",T.Kth(x));else if (c==5) printf("%d\n",T.Pred(x));else if (c==6) printf("%d\n",T.Succ(x)); } return 0; }

【非旋轉Treap】

#include<cstdio>
#include<cstdlib>
const int maxn=100005;
struct node{
    node *s[2];
    int k,f,siz;
    node () {}
    node (int w,node *son) {s[0]=s[1]=son;k=w;f=rand();siz=1;}
    void pushup() {siz=s[0]->siz+s[1]->siz+1;}
}nil,base[maxn];
typedef node* P_node;
P_node null,len,Root;
void treap_init(){
    nil=node(0,NULL);null=&nil;
    null->s[0]=null->s[1]=null;
    null->siz=0; len=base; Root=null;
}
P_node newnode(int x){
    *len=node(x,null);
    return len++;
}
P_node merge(P_node a,P_node b){
    if (a==null) return b;
    if (b==null) return a;
    if (a->f > b->f) {a->s[1]=merge(a->s[1],b); a->pushup(); return a;}
    b->s[0]=merge(a,b->s[0]); b->pushup(); return b;
}
void split(P_node x,int k,P_node &l,P_node &r){
    if (x==null) {l=r=null;return;}
    if (x->k <= k) {l=x;split(l->s[1],k,l->s[1],r);}else
     {r=x;split(r->s[0],k,l,r->s[0]);}
    x->pushup();
}
void Insert(int k){
    P_node l,r;
    split(Root,k,l,r);
    Root=merge(merge(l,newnode(k)),r);
}
void Erase(int k){
    P_node l,mid,r;
    split(Root,k-1,l,r);split(r,k,mid,r);
    Root=merge(l,merge(merge(mid->s[0],mid->s[1]),r));
}
int Kth(P_node x,int k){
    if (k <= x->s[0]->siz) return Kth(x->s[0],k);else
    if (k > x->s[0]->siz+1) return Kth(x->s[1],k - x->s[0]->siz -1);
    return x->k;
}
int get_edge(P_node x,int d){
    while (x->s[d]!=null) x=x->s[d];
    return x->k;
}
int q;
inline int red(){
    int tot=0,f=1;char ch=getchar();
    while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=getchar();}
    while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar();
    return tot*f;
}
int main(){
    treap_init();
    q=red();
    while (q--){
        int c=red();
        if (c==1) Insert(red());else
        if (c==2) Erase(red());else
        if (c==3){
            int k=red();P_node l,r;
            split(Root,k-1,l,r); printf("%d\n",l->siz+1); Root=merge(l,r);
        }else
        if (c==4) printf("%d\n",Kth(Root,red()));else
        if (c==5){
            int k=red();P_node l,r;
            split(Root,k-1,l,r); printf("%d\n",get_edge(l,1)); Root=merge(l,r);
        }else{
            int k=red();P_node l,r;
            split(Root,k,l,r); printf("%d\n",get_edge(r,0)); Root=merge(l,r);
        }
    }
    return 0;
}

相關推薦

Treap/旋轉TreapBZOJ3224 [Tyvj1728]普通平衡

題面在這裡 平衡樹裸題…… 寫了Treap/非旋轉Treap兩個版本。 【普通Treap】 #include<cstdio> #include<algorithm> using namespace std; const int

treapbzoj 3224: Tyvj 1728 普通平衡

加上了前驅和後繼 //#define _TEST _TEST #include <cstdio> #include <cstring> #include <cstdlib> #include <iostream> #inc

模板bzoj-3224普通平衡(splay&treap&SBT)

先吐槽一下,要找塊好模板真不容易,畢竟大家都有各自的程式碼風格,強迫自己去看非自己風格的程式碼真是痛苦,所以在網上新增一種型別的模板 最近才發現這題A了三遍,分別是以三種平衡樹的模板題A的,所以稍微彙總一下,讓後來者有個型別的模板可以依靠 相信看模板的人都看

bzoj 3224Tyvj 1728 普通平衡

Tyvj 1728 普通平衡樹 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 11767 Solved: 5021 [Submit][Status][Discuss] Description 您需要

模板旋轉Treap

Code #include <cstdio> #include <cstdlib> #include <iostream> #include <algorithm> #include <cmath> #include <

BZOJ1014 JSOI2008 火星人prefix 旋轉Treap*

BZOJ1014 JSOI2008 火星人prefix Description   火星人最近研究了一種操作:求一個字串兩個字尾的公共字首。比方說,有這樣一個字串:madamimadam,我們將這個字串的各個字元予以標號:序號: 1 2 3 4 5

Treap[BZOJ 3224]Tyvj 1728 普通平衡 & 旋轉實現

#include<cstdio> #include<algorithm> #include<cstring> using namespace std; #define maxn 2000005 struct Treap{

bzoj3224普通平衡——treap

blog lib mes cas style tree upd treap ins 我的第一道treap題目,treap的模版題。 代碼是對著hzw的敲的,一邊敲一邊理解。。。 主要是熟悉一下treap的各種基本操作,詳細細節看代碼。 #include<cstdio

紅黑 ------ luogu P3369 模板普通平衡Treap/SBT)

div child lin main false tchar clas char als 二次聯通門 : luogu P3369 【模板】普通平衡樹(Treap/SBT) 近幾天閑來無事。。。就把各種平衡樹都寫了一下。。。 下面是紅黑樹(Red Black Tree)

替罪羊 ------ luogu P3369 模板普通平衡Treap/SBT)

nod %d clas https number problem 普通 true ble 二次聯通門 : luogu P3369 【模板】普通平衡樹(Treap/SBT) 閑的沒事,把各種平衡樹都寫寫 比較比較。。。 下面是替罪羊樹 #include &l

數組splay ------ luogu P3369 模板普通平衡Treap/SBT)

普通 模板 char truct div color fine col suffix 二次聯通門 : luogu P3369 【模板】普通平衡樹(Treap/SBT) #include <cstdio> #define Max 100005

fhq treap ------ luogu P3369 模板普通平衡Treap/SBT)

ret true read std stdin urn tdi ref code 二次聯通門 : LibreOJ #104. 普通平衡樹 #include <cstdio> #include <iostream> #include

luogu3369 模板普通平衡Treap/SBT)

oid sin markdown 隨機 節點 nod -m rotate 編號 treap做法,參考hzwer的博客 #include <iostream> #include <cstdlib> #include <cstdio> usi

題解 P3369 模板普通平衡Treap/SBT)

為什麽 平衡樹 -m cout 指向 erase out 刪除 play STL真是個好東西。 最近在看pb_ds庫及vector和set的用法,就想用這三種操作來實現一下普通平衡樹,結果pb_ds中的rbtree不支持重復值,而本蒟蒻也看不懂不懂各大佬用pb_ds的實現,

[bzoj3173]最長上升子序列_旋轉Treap

向上 ont clas ring pos pan rand() names include 最長上升子序列 bzoj-3173     題目大意:有1-n,n個數,第i次操作是將i加入到原有序列中制定的位置,後查詢當前序列中最長上升子序列長度。     註釋:1<

模板普通平衡Treap/SBT)

gpo splay 格式 可能 bit bsp inf ron 表示 題目描述 您需要寫一種數據結構(可參考題目標題),來維護一些數,其中需要提供以下操作: 插入x數 刪除x數(若有多個相同的數,因只刪除一個) 查詢x數的排名(排名定義為比當前數小的數的個數+1

[BZOJ3224]普通平衡(旋轉treap)

gree 答案 需要 discus space scrip 其中 個數 names 3224: Tyvj 1728 普通平衡樹 Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 20328 Solved: 8979[Su

luogu P3369 模板普通平衡Treap/SBT)

AS += amp father https ota fin size insert 題目鏈接:https://www.luogu.org/problemnew/show/P3369 #include <cstdio> #include <algorith

BZOJ1500[NOI2005]維修數列——旋轉treap

char emp height src alt 格式 == bzoj https 題目描述 請寫一個程序,要求維護一個數列,支持以下 6 種操作: 請註意,格式欄 中的下劃線‘ _ ’表示實際輸入文件中的空格 輸入 輸入的第1

[bzoj1895][Pku3580]supermemo_旋轉Treap

小結 fin 一個點 output 翻轉 旋轉 pac ron 平移 supermemo bzoj-1895 Pku-3580 題目大意:給定一個n個數的序列,需支持:區間加,區間翻轉,區間平移,單點插入,單點刪除,查詢區間最小值。 註釋:$1\le n\le 6.1\