题目

你需要写一种数据结构支援以下操作。

  1. 插入元素。
  2. 删除元素。
  3. 查询元素的排名。
  4. 查询第k小的元素。
  5. 查询元素前趋。
  6. 查询元素后继。

题解

BBST裸题。

代码

#include <cstdio>
#include <iostream>
#define ll long long
using namespace std;
#ifdef D
const int maxn = 100;
#else
const int maxn = 100005;
#endif
int ch[maxn][2], fa[maxn], size[maxn], data[maxn], cnt[maxn];
int rt, n, m;
ll sz = 0;
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;
}
inline void update(int x) {
size[x] = size[ch[x][0]] + size[ch[x][1]] + cnt[x];
}
inline void zig(int x) {
int y = fa[x], z = fa[y], l = (ch[y][1] == x), r = l ^ 1;
fa[ch[y][l] = ch[x][r]] = y;
fa[ch[x][r] = y] = x;
fa[x] = z;
if (z)
ch[z][ch[z][1] == y] = x;
update(y);
}
inline void splay(int x, int aim = 0) { //调整x成为aim的儿子
for (int y; (y = fa[x]) != aim; zig(x)) {
if (fa[y] != aim)
zig((ch[y][0] == x) == (ch[fa[y]][0] == y) ? y : x);
}
if (aim == 0)
rt = x;
update(x);
}
inline void insert(int v) {
int x = rt;
if (rt == 0) {
rt = x = ++sz;
data[x] = v;
fa[x] = ch[x][0] = ch[x][1] = 0;
size[x] = cnt[x] = 1;
return;
}
while (x) {
size[x]++;
if (v == data[x]) {
cnt[x]++;
return;
}
int &y = ch[x][v >= data[x]];
if (y == 0) {
y = ++sz;
data[y] = v;
fa[y] = x;
ch[y][0] = ch[y][1] = 0;
size[y] = cnt[y] = 1;
x = y;
break;
}
x = y;
}
splay(x);
}
int rank(int v) {
int ret = 0, ans, x = rt;
while (x) {
if (v > data[x]) {
ret += size[ch[x][0]] + cnt[x];
x = ch[x][1];
} else {
if (v == data[x]) {
return ret + size[ch[x][0]] + 1;
}
x = ch[x][0];
}
}
}
int kth(int v) {
int x = rt;
while (x) {
if (v >= size[ch[x][0]] + 1 && v <= size[ch[x][0]] + cnt[x])
return data[x];
if (v > size[ch[x][0]] + cnt[x]) {
v -= size[ch[x][0]] + cnt[x];
x = ch[x][1];
} else
x = ch[x][0];
}
}
int pre(int v) {
int ans;
for (int y = rt; y;)
if (data[y] < v)
ans = data[y], y = ch[y][1];
else
y = ch[y][0];
return ans;
}
int succ(int v) {
int ans;
for (int y = rt; y;)
if (data[y] <= v)
y = ch[y][1];
else
ans = data[y], y = ch[y][0];
return ans;
}
inline void del(int v) {
int x = rt;
while (x) {
size[x]--;
if (data[x] == v) {
cnt[x]--;
break;
}
int &y = ch[x][v >= data[x]];
x = y;
}
if (cnt[x] != 0)
return;
splay(x);
if (ch[x][0] == 0) {
rt = ch[x][1];
fa[rt] = 0;
return;
}
if (ch[x][1] == 0) {
rt = ch[x][0];
fa[rt] = 0;
return;
}
x = ch[rt][0];
while (ch[x][1])
x = ch[x][1];
splay(x, rt);
ch[x][1] = ch[rt][1];
fa[ch[x][1]] = x;
size[x] = size[rt] - 1;
fa[x] = 0;
rt = x;
}
int main() {
#ifdef DEBUG
freopen("input", "r", stdin);
#endif
scanf("%d", &n);
rt = sz = 0;
while (n--) {
int opt = read();
if (opt == 1) {
int x = read();
insert(x); // O(log n)
} else if (opt == 2) {
int x = read();
del(x); // O(log n)
} else if (opt == 3) {
int k = read();
printf("%d\n", rank(k));
} else if (opt == 4) {
int k = read();
printf("%d\n", kth(k));
} else if (opt == 5) {
int x = read();
printf("%d\n", pre(x));
} else if (opt == 6) {
int x = read();
printf("%d\n", succ(x));
}
}
}

[bzoj3224]Tyvj 1728 普通平衡树——splay模板的更多相关文章

  1. bzoj3224: Tyvj 1728 普通平衡树(splay)

    3224: Tyvj 1728 普通平衡树 题目:传送门 题解: 啦啦啦啦又来敲个模版水经验啦~ 代码: #include<cstdio> #include<cstring> ...

  2. [BZOJ3224]Tyvj 1728 普通平衡树

    [BZOJ3224]Tyvj 1728 普通平衡树 试题描述 您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:1. 插入x数2. 删除x数(若有多个相同的数,因只删除一个) ...

  3. bzoj3224: Tyvj 1728 普通平衡树(平衡树)

    bzoj3224: Tyvj 1728 普通平衡树(平衡树) 总结 a. cout<<(x=3)<<endl;这句话输出的值是3,那么对应的,在splay操作中,当父亲不为0的 ...

  4. 【BZOJ3224】Tyvj 1728 普通平衡树 Splay

    Description 您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:1. 插入x数2. 删除x数(若有多个相同的数,因只删除一个)3. 查询x数的排名(若有多个相同的数 ...

  5. bzoj3224 Tyvj 1728 普通平衡树(名次树+处理相同)

    3224: Tyvj 1728 普通平衡树 Time Limit: 10 Sec  Memory Limit: 128 MBSubmit: 5354  Solved: 2196[Submit][Sta ...

  6. BZOJ3224 洛谷3369 Tyvj 1728 普通平衡树 splay

    欢迎访问~原文出处——博客园-zhouzhendong 去博客园看该题解 题目传送门 - BZOJ3224 题意概括 您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:1. ...

  7. 绝对是全网最好的Splay 入门详解——洛谷P3369&BZOJ3224: Tyvj 1728 普通平衡树 包教包会

    平衡树是什么东西想必我就不用说太多了吧. 百度百科: 一个月之前的某天晚上,yuli巨佬为我们初步讲解了Splay,当时接触到了平衡树里的旋转等各种骚操作,感觉非常厉害.而第二天我调Splay的模板竟 ...

  8. bzoj3224: Tyvj 1728 普通平衡树(打个splay暖暖手)

    (其实今天好热啊? 题目大意:插入,删除,k小,前驱后继,数的排名. splay和treap裸题...过几天补个treap的 splay: #include<iostream> #incl ...

  9. 【Splay】bzoj3224 Tyvj 1728 普通平衡树

    #include<cstdio> #include<iostream> #include<cstring> #include<algorithm> us ...

随机推荐

  1. 手把手教你玩转CSS3 3D技术

    手把手教你玩转 CSS3 3D 技术   要玩转css3的3d,就必须了解几个词汇,便是透视(perspective).旋转(rotate)和移动(translate).透视即是以现实的视角来看屏幕上 ...

  2. idea在Maven Projects中显示灰色的解决办法

    问题描述: 在使用idea的过程中,遇到其中一个maven模块变成灰色,如下所示: 问题解决: 造成这个的原因可能是忽略了maven模块. 可以尝试如下解决方法:在idea中进入Settings–&g ...

  3. C++学习003-#define 自定义宏

    C++中可以用#define来定义自定义的宏 也可以用使用#define来定义常量 但是#define只是简单的替换,在定义常量的时候没有语法检测 所以在C++定义常量可以使用  Const修饰 #d ...

  4. CSS3 :animation 动画

    CSS3动画分为二部份: 1.定义动画行为: 使用@keyframes定义动画行为,有两种方式: 方式一:仅定义动画起始样式,与动画结束样式 @keyframes (动画行为名称) { from {b ...

  5. python 学习总结----正则表达式

    正则表达式 应用场景 - 特定规律字符串的查找,切割,替换 - 邮箱格式:URl,IP地址等的校验 - 爬虫项目中,特定内容的提取 使用原则 - 只要使用字符串等函数能解决的问题,就不要使用正则 - ...

  6. Java IO学习--输入输出流

    一.Java IO的定义 I/O:输入输出系统,由输入输出控制系统和外围设备两部分组成. Java中I/O操作主要是指使用Java进行输入,输出操作. Java所有的I/O机制都是基于数据流进行输入输 ...

  7. Leetcode. 回文字符串的分割和最少分割数

    Q1: 回文字符串的分割 Given a string s, partition s such that every substring of the partition is a palindrom ...

  8. POJ 1149 PIGS(最大流)

    Description Mirko works on a pig farm that consists of M locked pig-houses and Mirko can't unlock an ...

  9. PAT 1045 快速排序

    https://pintia.cn/problem-sets/994805260223102976/problems/994805278589960192 著名的快速排序算法里有一个经典的划分过程:我 ...

  10. 【考试记录】4.8 Path (网络流 —— 劲题)

    手抄代码 + 学习指针 + 冥思苦想一晚上终于——在一瞬间开窍了.果然题目都是这样:突破了一个点,一切都是柳暗花明. 题面描述: 样例: 这道题目,首先注意到给定的边的性质:这些边在平面上构成了一棵树 ...