BZOJ3224:普通平衡树(含SBT、Treap、Splay模板)

最新推荐文章于 2019-08-14 17:24:49 发布
最新推荐文章于 2019-08-14 17:24:49 发布
阅读量3.2k 收藏 1
点赞数
分类专栏： 平衡树-区间树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/gengmingrui/article/details/49978543
版权

传送门
这道题就是裸的平衡树的题目，操作都是模板，也就不多说了，重在代码。

SBT:

#include<cstdio>
struct SBT {
    int v, sz, ch[2], cnt;
} t[300005];
int n, cnt, root;
#define Upd(k) {\
    t[k].sz = t[t[k].ch[0]].sz + t[t[k].ch[1]].sz + t[k].cnt;\
}
void rot(int &k, bool f) {
    int p = t[k].ch[f]; t[k].ch[f] = t[p].ch[!f]; t[p].ch[!f] = k;
    Upd(k); Upd(p); k = p;
}
inline void mt(int &k,bool f)
{
    if(!k) return;
    if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f]].sz) rot(k, f);
    else if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f^1]].sz) { rot(t[k].ch[f], f^1); rot(k, f); }
    else return;
    mt(t[k].ch[f],f);
    mt(k,f);
}
void Ins(int &k, int x) {
    if(!k) {k = ++ cnt; t[k].sz = t[k].cnt = 1; t[k].v = x; return; }
    ++ t[k].sz;
    if(t[k].v == x) {++ t[k].cnt; return;}
    Ins(t[k].ch[t[k].v < x], x);
    mt(k, t[k].v < x);
}
int Del(int &k, int x) {
    if(!k) return k;
    int tmp;
    if(t[k].v == x) {
        if(t[k].cnt > 1) {-- t[k].cnt; --t[k].sz; return k;}
        else if(!(t[k].ch[0]*t[k].ch[1])) {k = t[k].ch[0]+t[k].ch[1];}
        else { tmp = Del(t[k].ch[0], x+1); t[k].cnt = t[tmp].cnt; t[k].v = t[tmp].v; Upd(k); return k; }
    }
    else if((t[k].v < x && !t[k].ch[1]) || (t[k].v > x && ! t[k].ch[0])) { tmp = k; k = t[k].ch[0]; Upd(k); return tmp; }
    else tmp = Del(t[k].ch[t[k].v < x], x);
    Upd(k); return tmp;
}
int Ran(int k,int x) {
    if(k==0)return 0;
    if(t[k].v==x)return t[t[k].ch[0]].sz+1;
    else if(x>t[k].v)
        return t[t[k].ch[0]].sz+t[k].cnt+Ran(t[k].ch[1],x);
    else return Ran(t[k].ch[0],x);
}
int Kth(int k,int x) {
    if(k==0)return 0;
    if(x<=t[t[k].ch[0]].sz)
        return Kth(t[k].ch[0],x);
    else if(x>t[t[k].ch[0]].sz+t[k].cnt)
        return Kth(t[k].ch[1],x-t[t[k].ch[0]].sz-t[k].cnt);
    else return t[k].v;
}
int ans;
void pred(int k,int x) {
    if(k==0)return;
    if(t[k].v<x) {
        ans=k;
        pred(t[k].ch[1],x);
    } else pred(t[k].ch[0],x);
}
void succ(int k,int x) {
    if(k==0)return;
    if(t[k].v>x) {
        ans=k;
        succ(t[k].ch[0],x);
    } else succ(t[k].ch[1],x);
}
char c, f;
inline void GET(int &n) {
    n = 0; f = 1;
    do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0');
    while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();}
    n *= f;
}
int main() {
    GET(n);
    int opt,x;
    for(int i=1; i<=n; i++) {
        GET(opt); GET(x);
        switch(opt) {
            case 1: Ins(root,x); break;
            case 2: Del(root,x); break;
            case 3: printf("%d\n",Ran(root,x)); break;
            case 4: printf("%d\n",Kth(root,x)); break;
            case 5: ans=0; pred(root,x); printf("%d\n",t[ans].v); break;
            case 6: ans=0; succ(root,x); printf("%d\n",t[ans].v); break;
        }
    }
    return 0;
}

Treap:

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
struct data {
    int ch[2], v, sz, rnd, cnt;
} tr[300005];
int n,size,root,ans,sd = 2333;
inline int ran() { return sd = (sd*sd+sd)%(10003); }
inline void Upd(int k) {
    tr[k].sz = tr[tr[k].ch[0]].sz + tr[tr[k].ch[1]].sz + tr[k].cnt;
}
void rot(int &k, bool f) { ///0->right_rotate
    int t = tr[k].ch[f]; tr[k].ch[f] = tr[t].ch[!f]; tr[t].ch[!f] = k;
    Upd(k); Upd(t); k = t;
}
void insert(int &k,int x) {
    if(k==0) { size++; k=size; tr[k].sz=tr[k].cnt=1; tr[k].v=x; tr[k].rnd=ran(); return; }
    tr[k].sz++;
    if(tr[k].v==x) { tr[k].cnt++; return; }
    bool f = x>tr[k].v;
    insert(tr[k].ch[f],x);
    if(tr[tr[k].ch[f]].rnd<tr[k].rnd)rot(k, f);
}

void del(int &k,int x) {
    if(k==0)return;
    if(tr[k].v==x) {
        if(tr[k].cnt>1) { tr[k].cnt--; tr[k].sz--; return; }
        if(tr[k].ch[0]*tr[k].ch[1]==0)k=tr[k].ch[0]+tr[k].ch[1];
        else rot(k, tr[tr[k].ch[0]].rnd >= tr[tr[k].ch[1]].rnd), del(k, x);
    } else tr[k].sz--, del(tr[k].ch[x>tr[k].v], x);
}
int get_rank(int k,int x) {
    if(k==0)return 0;
    if(tr[k].v==x)return tr[tr[k].ch[0]].sz+1;
    else if(x>tr[k].v)
        return tr[tr[k].ch[0]].sz+tr[k].cnt+get_rank(tr[k].ch[1],x);
    else return get_rank(tr[k].ch[0],x);
}
int get_kth(int k,int x) {
    if(k==0)return 0;
    if(x<=tr[tr[k].ch[0]].sz)
        return get_kth(tr[k].ch[0],x);
    else if(x>tr[tr[k].ch[0]].sz+tr[k].cnt)
        return get_kth(tr[k].ch[1],x-tr[tr[k].ch[0]].sz-tr[k].cnt);
    else return tr[k].v;
}
void pred(int k,int x) {
    if(k==0)return;
    if(tr[k].v<x) {
        ans=k;
        pred(tr[k].ch[1],x);
    } else pred(tr[k].ch[0],x);
}
void succ(int k,int x) {
    if(k==0)return;
    if(tr[k].v>x) {
        ans=k;
        succ(tr[k].ch[0],x);
    } else succ(tr[k].ch[1],x);
}char c, f;
inline void GET(int &n) {
    n = 0; f = 1;
    do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0');
    while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();}
    n *= f;
}
int main() {
    GET(n);
    int opt,x;
    for(int i=1; i<=n; i++) {
        GET(opt); GET(x);
        switch(opt) {
            case 1: insert(root,x); break;
            case 2: del(root,x); break;
            case 3: printf("%d\n",get_rank(root,x)); break;
            case 4: printf("%d\n",get_kth(root,x)); break;
            case 5: ans=0; pred(root,x); printf("%d\n",tr[ans].v); break;
            case 6: ans=0; succ(root,x); printf("%d\n",tr[ans].v); break;
        }
    }
    return 0;
}

Splay

#include<cstdio>
#define MAXN 300005
struct node {
    int v, cnt, sz, ch[2], f;
}t[MAXN];
int rt, sz, n;
#define Upd(r) {t[r].sz = t[t[r].ch[0]].sz + t[t[r].ch[1]].sz + t[r].cnt;}
void rot(int x)
{
    int y = t[x].f, z = t[y].f;
    bool f = (t[y].ch[1] == x);
    t[y].ch[f] = t[x].ch[f^1];
    if(t[y].ch[f]) t[t[y].ch[f]].f = y;
    t[x].ch[f^1] = y; t[y].f = x;
    t[x].f = z;
    if(z) t[z].ch[t[z].ch[1]==y] = x;
    Upd(y);
}
void Spaly(int r, int tp) {
    for(int y, z; (y = t[r].f) != tp; rot(r)) {
        z = t[y].f;
        if(z == tp) continue;
        if( (t[z].ch[0] == y) == (t[y].ch[0] == r) ) rot(y);
        else rot(r);
    }
    if(!tp) rt = r; Upd(r);
}
void Ins(int r, int x) {
    int y = 0;
    while(r && t[r].v != x) { y = r; r = t[r].ch[x > t[r].v]; }
    if(r) ++ t[r].cnt;
    else {
        r = ++ sz; t[r].sz = t[r].cnt = 1;
        t[r].v = x; t[r].f = y; if(y) t[y].ch[x > t[y].v] = r;
    }
    Spaly(r, 0);
}
void Find(int v) {
    int x = rt; if(!x) return;
    while(t[x].ch[v > t[x].v] && t[x].v != v) x = t[x].ch[v > t[x].v];
    Spaly(x, 0);
}
int Ran(int v) {
    Find(v);
    return t[t[rt].ch[0]].sz;
}
int Kth(int x)
{
    int y=rt,p;
    if(x>t[rt].sz)return 0;
    while(1)
    {
        p=t[y].ch[0];
        if(t[p].sz+t[y].cnt<x) {
            x-=t[p].sz+t[y].cnt;
            y=t[y].ch[1];
        }
        else if(t[p].sz>=x) y=p;
        else return t[y].v;
    }
}
int Nxt(int x, bool f)
{
    Find(x);
    if((t[rt].v>x&&f)||(t[rt].v<x&&!f)) return rt;
    int p = t[rt].ch[f];
    while(t[p].ch[f^1]) p = t[p].ch[!f];
    return p;
}
void Del(int v) {
    int p = Nxt(v, 0), s = Nxt(v, 1);
    Spaly(p, 0); Spaly(s, p);
    p = t[s].ch[0];
    if(t[p].cnt > 1) -- t[p].cnt, Spaly(p, 0);
    else t[s].ch[0] = 0;
}
char c, f;
inline void GET(int &n) {
    n = 0; f = 1;
    do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0');
    while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();}
    n *= f;
}
int main() {
    GET(n);
    int opt,x;
    Ins(rt, -0x7fffffff); Ins(rt, +0x7fffffff);
    for(int i=1; i<=n; i++) {
        GET(opt); GET(x);
        switch(opt) {
            case 1: Ins(rt,x); break;
            case 2: Del(x); break;
            case 3: printf("%d\n",Ran(x)); break;
            case 4: printf("%d\n",Kth(x+1)); break;
            case 5: printf("%d\n",t[Nxt(x, 0)].v); break;
            case 6: printf("%d\n",t[Nxt(x, 1)].v); break;
        }
    }
    return 0;
}
（vector通过）bzoj3224: Tyvj 1728 普通平衡树
.
08-18 1万+
3224: Tyvj 1728 普通平衡树 Time Limit:10 SecMemory Limit:128 MB Submit:15204Solved:6614 [Submit][Status][Discuss] Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的
【BZOJ3224】【TYVJ1728】普通平衡树
cz_xuyixuan的博客
01-14 662
【题目链接】 点击打开链接 【思路要点】 本题包含了平衡树最基本的操作。是任何学习平衡树都应当先做一遍的题。笔者实现了四种平衡树，Splay、Treap、替罪羊树和非旋转式Treap（以及其可持久化）。 【代码】 Splay/*Splay Tree Version*/ #include using namespace std;
bzoj 3224（平衡树）
Chen.Jr的博客
03-21 222
传送门 题意： 请你完成666种最基础的平衡树的操作。 题目分析： 什么也别说了，平衡树的板子题。收获到了一个写起来比较舒服的平衡树的板子。 代码： #include <bits/stdc++.h> using namespace std; const int maxn=200005; int ch[maxn][2],par[maxn],val[maxn],cnt[maxn],siz...
bzoj3224
weixin_30500289的博客
04-05 105
Tyvj 1728 普通平衡树 HYSBZ - 3224 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：1. 插入x数2. 删除x数(若有多个相同的数，因只删除一个)3. 查询x数的排名(若有多个相同的数，因输出最小的排名)4. 查询排名为x的数5. 求x的前驱(前驱定义为小于x，且最大的数)6. 求x的后继(后继定义为大于x，且最小的数) ...
【bzoj3224】普通平衡树
zyf2000
02-04 4965
如何让你遇见我，在我最美丽的时刻。
bzoj3223(文艺平衡树 splay区间翻转模板)
qq_42576687的博客
08-14 212
题目 思路： 以下标建立一颗平衡树。splay之后序列顺序并不改变，就像splay之后不改变数的大小关系。翻转区间[l,r]先把 l-1 翻转到根部，然后把r+1翻转到根部的右儿子。此时根部右儿子的左儿子就是区间[l,r]。此时对这个区间tag[ch[ch[root][1]][0]]^=1; 每次route的时候需要pushdown(f[x]),pushdown(x). 将l-1旋转到根部的操作是...
SBT模板(平衡树)
04-02
本模板为 BZOJ3224：文艺平衡树 的源程序 含各种操作，旋转，插入，删除，求前驱，后继，查询值为x的数的排名，查询排名为k的数，求最大值，最小值……
bzoj 3224: Tyvj 1728 普通平衡树
lqybzx的专栏
11-05 696
Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除一个) 3. 查询x数的排名(若有多个相同的数，因输出最小的排名) 4. 查询排名为x的数 5. 求x的前驱(前驱定义为小于x，且最大的数) 6. 求x的后继(后继定义为大于x，且最小的数) Input 第
【模板】SBT 普通平衡树
高斯的博客
11-24 2510
平衡树
【模板】普通平衡树（Treap/SBT）
Brute♂force
06-04 394
题目描述 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 插入 xx 数 删除 xx 数(若有多个相同的数，因只删除一个) 查询 xx 数的排名(排名定义为比当前数小的数的个数 +1+1 。若有多个相同的数，因输出最小的排名) 查询排名为 xx 的数 求 xx 的前驱(前驱定义为小于 xx ，且最大的数) 求 xx 的后继(后继定义为大于 xx ，且最小...
【BZOJ3224】普通平衡树
Facico的博客
07-28 976
Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除一个) 3. 查询x数的排名(若有多个相同的数，因输出最小的排名) 4. 查询排名为x的数 5. 求x的前驱(前驱定义为小于x，且最大的数) 6. 求x的后继(后继定义为大于x，且最小的数) Solution 这是一道练习
BZOJ 3224 普通平衡树
我是谁 --- 卡比兽
09-26 194
BZOJ 3224 懒羊羊看了下BST treap  然后对着敲一下BZOJ 3224 代码有注释 额 感觉treap好玄学啊 rand是什么鬼 #include &lt;cstdio&gt; #include &lt;iostream&gt; #include &lt;queue&gt; #include &lt;cmath&gt; #include &lt;cstring&gt; #...
bzoj3224(普通平衡树 splay板子)
qq_42576687的博客
07-26 173
题目 splay #include<cstdio> #include<iostream> using namespace std; const int N=1e5+5; int ch[N][2],f[N],sz[N],cnt[N],val[N]; int tot,root; inline void _clear(int x){ ch[x][0]=ch[x][1]=...
【BZOJ 3224】普通平衡树
weixin_34248705的博客
05-28 84
【原题题面】传送门 【题解大意】 平衡树模板题。 左旋右旋使数尽量满足左右子树相当的情况。 随机化权值是因为在随机数据下BST趋近于平衡。 同时要维护子树大小和样本容量来回答询问。 具体实现看代码。 【code】 #include<bits/stdc++.h> using namespace std; #define File1 "input" #defi...
[bzoj3224]普通平衡树
jerome_wei的博客
10-13 226
[bzoj3224]普通平衡树 平衡树板子。因为懒得删点，这里直接num--，然后要注意查询前驱后继的时候不能直接输出只能（我只会）再查一遍Kth 写得丑别打我 代码 #include&lt;bits/stdc++.h&gt; using namespace std; const int N=1e5+5; struct Splay{ int ch[2],fa; int ...
BZOJ2194: 快速傅立叶之二 (FFT)
最新发布
05-12
这是一道经典的FFT模板题目，需要你掌握FFT的基本原理和实现。 FFT（快速傅里叶变换）是一种用于计算离散傅里叶变换（DFT）的算法。它将DFT的时间复杂度从O(N^2)降低到了O(N*logN)，因此在处理大规模数据时具有很高...
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值