题目描述
创建带索引的二叉搜索树类。存储结构使用链表,提供操作:插入、删除、按名次删除、查找、按名次查找、升序输出所有元素。
输入输出格式
输入格式
输入第一行一个数字 m (m<=1000000),表示有 m 个操作。
接下来 m 行,每一行有两个数字 a,b:
- 当输入的第一个数字 a 为 0 时,输入的第二个数字 b 表示向搜索树中插入 b
- 当输入的第一个数字 a 为 1 时,输入的第二个数字 b 表示向搜索树中查找 b
- 当输入的第一个数字 a 为 2 时,输入的第二个数字 b 表示向搜索树中删除 b
- 当输入的第一个数字 a 为 3 时,输入的第二个数字 b 表示查找搜索树中名次为 b 的元素
- 当输入的第一个数字 a 为 4 时,输入的第二个数字 b 表示删除搜索树中名次为 b 的元素
输出格式
对于输入中的每一种操作,输出执行操作的过程中依次比较的元素值的异或值。
注意
- 查询与删除操作中,待查询的元素也需要异或入答案中
- 查找(删除)操作中,如果未找到,或者插入操作,已存在,输出 0(不插入),不需要输出异或和
- 查找(删除)第 b 大,如果不存在,输出 0
- 删除操作中,如果当前元素有两个孩子,替换的为 右子树中最小的,如果只有一个孩子,直接用该孩子替换当前元素,如果没有孩子,直接删除
- 删除操作的替换过程中所有比较操作不计入答案
数据结构与算法描述
构造带索引的二叉搜索树节点的结构体,成员有元素、左子树节点数、左右孩子、构造函数。
struct indexedBStreeNode {
int element;//元素
int leftsize;//左子树节点数
indexedBStreeNode* leftchild;
indexedBStreeNode* rightchild;
indexedBStreeNode(int a) {
element = a;
leftsize = 0;
leftchild = NULL;
rightchild = NULL;
}
};
构造带索引的二叉搜索树类,私有成员为跟几点、树的大小。公有成员有插入、删除、按名次删除、查找、按名次查找函数。
class indexedBStree {
private:
indexedBStreeNode* root;
int size;//树的大小
public:
indexedBStree() {
root = NULL;
size = 0;
}
void insert(int b);//插入
void find(int b);//查找b
void erase(int b);//删除b
void get(int index);//查找名次为b的元素
void idelete(int index);//删除名次为b的元素
};
插入函数:声明一个初始化为根节点的搜索指针,和一个搜索指针的父节点的指针。插入的大则向右子树,插入的小则向左子树。找到节点后,如果树为空则新节点在根上,其他的插在搜索指针的父节点的子节点上。再更新左子树的节点个数,即如果插入的树小于该节点元素说明在这个节点的左子树,所以该节点的左子树个数要加一。
void indexedBStree::insert(int b) {//插入
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点的指针
int result = 0;
while (p != NULL) {
pp = p;//pp向下移动
result ^= p->element;
if (b < p->element) {
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
else {//已经存在输出0
cout << 0 << endl;
return;
}
}
indexedBStreeNode* newNode = new indexedBStreeNode(b);//为插入的元素建立新的节点 ,作为pp的子节点
if (root != NULL) {//树非空
if (b < pp->element)pp->leftchild = newNode;
else pp->rightchild = newNode;
}
else {
root = newNode;
}
size++;
p = root;
while (p->element!=b) {//更新左子树的节点个数
if (b < p->element) {//如果插入的元素比该节点元素小,说明其在该节点的左子树,leftsize++
p->leftsize++;
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
}
cout << result << endl;
}
查找函数:从跟开始搜索,查找元素大则向右子树,插入元素小则向左子树。
void indexedBStree::find(int b) {//查找b
int result = 0;
indexedBStreeNode* p=root;//从根开始搜索
while (p != NULL) {
result ^= p->element;
if (b < p->element) {
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
else {//找到则返回异或值
cout << result << endl;
return;
}
}
cout << 0 << endl;//没找到
}
删除元素:和插入函数类似,找到要删除的节点p和其父节点pp。然后更新左子树节点个数。如果有两个孩子要转换成有0个或者1个孩子的情况。要找右子树的最小元素s,以及s的父节点ps,然后声明一个可以替代p指针的q指针,让q指针作为pp的孩子,让pp指向要删除节点(s)的父节点,将删除节点替换为s,删除节点两个孩子的情况改为0个或1个孩子的情况。保存删除节点的孩子指针,然后将孩子指针连接上pp。
void indexedBStree::erase(int b) {//删除b
int result = 0;
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点的指针
while (p != NULL && p->element != b) {//找到元素b所在的位置
pp = p;
if (b < p->element) {
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
if (p == NULL) {//不存在该元素
cout << 0 << endl;
return;
}
p = root;
while (p != NULL && p->element != b) {//更新左子树节点个数
result ^= p->element;
if (b < p->element) {
p->leftsize--;
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
result ^= p->element;
if (p->leftchild != NULL && p->rightchild != NULL) {//有两个孩子,转换成有0或1个孩子
indexedBStreeNode* s = p->rightchild;//找p的右子树的最小元素
indexedBStreeNode* ps = p;//s的父节点
while (s->leftchild != NULL) {//小元素在左孩子处
s->leftsize--;//更新左子树节点个数
ps = s;
s = s->leftchild;
}
indexedBStreeNode* q=new indexedBStreeNode(s->element);//声明一个可以替代p指针的q指针
q->leftsize = p->leftsize;
q->leftchild = p->leftchild;
q->rightchild = p->rightchild;
//让q指针作为pp的孩子
if (pp == NULL) {
root = q;
}
else if (p == pp->leftchild) {
pp->leftchild = q;
}
else {
pp->rightchild = q;
}
//让pp指向要删除节点(s)的父节点
if (ps == p) {
pp = q;
}
else {
pp = ps;
}
delete p;
p = s;//删除节点替换为s
//删除节点两个孩子的情况改为0个或1个孩子的情况
}
indexedBStreeNode* c;//保存删除节点的孩子指针
if (p->leftchild != NULL) {
c = p->leftchild;
}
else {
c = p->rightchild;
}
//将孩子指针连接上
if (p == root) {
root = c;
}
else {
if (p == pp->leftchild) {
pp->leftchild = c;
}
else {
pp->rightchild = c;
}
}
size--;//树的节点数减一
delete p;
cout << result << endl;
}
按索引查找元素:索引减一才表示左子树的节点个数,索引小于左子树节点个数时,说明该节点在左子树;大于的时候,在右子树,更新索引为原索引减去左子树个数和根。
void indexedBStree::get(int index) {
index--;//索引减一才表示左子树的节点个数
int result = 0;
indexedBStreeNode* p = root;
while (p != NULL && p->leftsize != index) {
result ^= p->element;
if (index < p->leftsize) {//索引小于左子树节点个数时,说明该节点在左子树
p = p->leftchild;
}
else {//大于的时候,在右子树
index -= (p->leftsize + 1);//在右子树,更新索引,原索引减去左子树个数和根
p = p->rightchild;
}
}
if (p == NULL) {//没找到
cout << 0 << endl;
return;
}
else {
result ^= p->element;
cout << result << endl;
}
}
按索引删除元素:先按索引查找,同时找到查找到的节点的父节点,然后就知道了该索引位置元素的大小,剩余和删除函数类似。
void indexedBStree::idelete(int index) {
index--;//索引减一才表示左子树的节点个数
int result = 0;
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点
while (p != NULL && p->leftsize != index) {
pp = p;
result ^= p->element;
if (index < p->leftsize) {//索引小于左子树节点个数时,说明该节点在左子树
p = p->leftchild;
}
else {//大于的时候,在右子树
index -= (p->leftsize + 1);//在右子树,更新索引,原索引减去左子树个数和根
p = p->rightchild;
}
}
if (p == NULL) {//没找到
cout << 0 << endl;
return;
}
result ^= p->element;
int b = p->element;//索引的位置所对应的元素,剩下的步骤和erase函数类似
p = root;
while (p != NULL && p->element != b) {//更新leftsize
if (b < p->element) {
p->leftsize--;
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
if (p->leftchild != NULL && p->rightchild != NULL) {//有两个孩子,转换成有0或1个孩子
indexedBStreeNode* s = p->rightchild;//找p的右子树的最小元素
indexedBStreeNode* ps = p;//s的父节点
while (s->leftchild != NULL) {//小元素在左孩子处
s->leftsize--;//更新左子树节点个数
ps = s;
s = s->leftchild;
}
indexedBStreeNode* q = new indexedBStreeNode(s->element);//声明一个可以替代p指针的q指针
q->leftsize = p->leftsize;
q->leftchild = p->leftchild;
q->rightchild = p->rightchild;
//让q指针作为pp的孩子
if (pp == NULL) {
root = q;
}
else if (p == pp->leftchild) {
pp->leftchild = q;
}
else {
pp->rightchild = q;
}
//让pp指向要删除节点(s)的父节点
if (ps == p) {
pp = q;
}
else {
pp = ps;
}
delete p;
p = s;//删除节点替换为s
//删除节点两个孩子的情况改为0个或1个孩子的情况
}
indexedBStreeNode* c;//保存删除节点的孩子指针
if (p->leftchild != NULL) {
c = p->leftchild;
}
else {
c = p->rightchild;
}
//将孩子指针连接上
if (p == root) {
root = c;
}
else {
if (p == pp->leftchild) {
pp->leftchild = c;
}
else {
pp->rightchild = c;
}
}
size--;//树节点树减一
delete p;
cout << result << endl;
}
完整代码(含注释)
#include<iostream>
#include<stdio.h>
using namespace std;
struct indexedBStreeNode {
int element;//元素
int leftsize;//左子树节点数
indexedBStreeNode* leftchild;
indexedBStreeNode* rightchild;
indexedBStreeNode(int a) {
element = a;
leftsize = 0;
leftchild = NULL;
rightchild = NULL;
}
};
class indexedBStree {
private:
indexedBStreeNode* root;
int size;//树的大小
public:
indexedBStree() {
root = NULL;
size = 0;
}
void insert(int b);//插入
void find(int b);//查找b
void erase(int b);//删除b
void get(int index);//查找名次为b的元素
void idelete(int index);//删除名次为b的元素
};
void indexedBStree::insert(int b) {//插入
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点的指针
int result = 0;
while (p != NULL) {
pp = p;//pp向下移动
result ^= p->element;
if (b < p->element) {
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
else {//已经存在输出0
cout << 0 << endl;
return;
}
}
indexedBStreeNode* newNode = new indexedBStreeNode(b);//为插入的元素建立新的节点 ,作为pp的子节点
if (root != NULL) {//树非空
if (b < pp->element)pp->leftchild = newNode;
else pp->rightchild = newNode;
}
else {
root = newNode;
}
size++;
p = root;
while (p->element!=b) {//更新左子树的节点个数
if (b < p->element) {//如果插入的元素比该节点元素小,说明其在该节点的左子树,leftsize++
p->leftsize++;
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
}
cout << result << endl;
}
void indexedBStree::find(int b) {//查找b
int result = 0;
indexedBStreeNode* p=root;//从根开始搜索
while (p != NULL) {
result ^= p->element;
if (b < p->element) {
p = p->leftchild;
}
else if (b > p->element) {
p = p->rightchild;
}
else {//找到则返回异或值
cout << result << endl;
return;
}
}
cout << 0 << endl;//没找到
}
void indexedBStree::erase(int b) {//删除b
int result = 0;
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点的指针
while (p != NULL && p->element != b) {//找到元素b所在的位置
pp = p;
if (b < p->element) {
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
if (p == NULL) {//不存在该元素
cout << 0 << endl;
return;
}
p = root;
while (p != NULL && p->element != b) {//更新左子树节点个数
result ^= p->element;
if (b < p->element) {
p->leftsize--;
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
result ^= p->element;
if (p->leftchild != NULL && p->rightchild != NULL) {//有两个孩子,转换成有0或1个孩子
indexedBStreeNode* s = p->rightchild;//找p的右子树的最小元素
indexedBStreeNode* ps = p;//s的父节点
while (s->leftchild != NULL) {//小元素在左孩子处
s->leftsize--;//更新左子树节点个数
ps = s;
s = s->leftchild;
}
indexedBStreeNode* q=new indexedBStreeNode(s->element);//声明一个可以替代p指针的q指针
q->leftsize = p->leftsize;
q->leftchild = p->leftchild;
q->rightchild = p->rightchild;
//让q指针作为pp的孩子
if (pp == NULL) {
root = q;
}
else if (p == pp->leftchild) {
pp->leftchild = q;
}
else {
pp->rightchild = q;
}
//让pp指向要删除节点(s)的父节点
if (ps == p) {
pp = q;
}
else {
pp = ps;
}
delete p;
p = s;//删除节点替换为s
//删除节点两个孩子的情况改为0个或1个孩子的情况
}
indexedBStreeNode* c;//保存删除节点的孩子指针
if (p->leftchild != NULL) {
c = p->leftchild;
}
else {
c = p->rightchild;
}
//将孩子指针连接上
if (p == root) {
root = c;
}
else {
if (p == pp->leftchild) {
pp->leftchild = c;
}
else {
pp->rightchild = c;
}
}
size--;//树的节点数减一
delete p;
cout << result << endl;
}
void indexedBStree::get(int index) {
index--;//索引减一才表示左子树的节点个数
int result = 0;
indexedBStreeNode* p = root;
while (p != NULL && p->leftsize != index) {
result ^= p->element;
if (index < p->leftsize) {//索引小于左子树节点个数时,说明该节点在左子树
p = p->leftchild;
}
else {//大于的时候,在右子树
index -= (p->leftsize + 1);//在右子树,更新索引,原索引减去左子树个数和根
p = p->rightchild;
}
}
if (p == NULL) {//没找到
cout << 0 << endl;
return;
}
else {
result ^= p->element;
cout << result << endl;
}
}
void indexedBStree::idelete(int index) {
index--;//索引减一才表示左子树的节点个数
int result = 0;
indexedBStreeNode* p = root;//搜索指针
indexedBStreeNode* pp = NULL;//p的父节点
while (p != NULL && p->leftsize != index) {
pp = p;
result ^= p->element;
if (index < p->leftsize) {//索引小于左子树节点个数时,说明该节点在左子树
p = p->leftchild;
}
else {//大于的时候,在右子树
index -= (p->leftsize + 1);//在右子树,更新索引,原索引减去左子树个数和根
p = p->rightchild;
}
}
if (p == NULL) {//没找到
cout << 0 << endl;
return;
}
result ^= p->element;
int b = p->element;//索引的位置所对应的元素,剩下的步骤和erase函数类似
p = root;
while (p != NULL && p->element != b) {//更新leftsize
if (b < p->element) {
p->leftsize--;
p = p->leftchild;
}
else {
p = p->rightchild;
}
}
if (p->leftchild != NULL && p->rightchild != NULL) {//有两个孩子,转换成有0或1个孩子
indexedBStreeNode* s = p->rightchild;//找p的右子树的最小元素
indexedBStreeNode* ps = p;//s的父节点
while (s->leftchild != NULL) {//小元素在左孩子处
s->leftsize--;//更新左子树节点个数
ps = s;
s = s->leftchild;
}
indexedBStreeNode* q = new indexedBStreeNode(s->element);//声明一个可以替代p指针的q指针
q->leftsize = p->leftsize;
q->leftchild = p->leftchild;
q->rightchild = p->rightchild;
//让q指针作为pp的孩子
if (pp == NULL) {
root = q;
}
else if (p == pp->leftchild) {
pp->leftchild = q;
}
else {
pp->rightchild = q;
}
//让pp指向要删除节点(s)的父节点
if (ps == p) {
pp = q;
}
else {
pp = ps;
}
delete p;
p = s;//删除节点替换为s
//删除节点两个孩子的情况改为0个或1个孩子的情况
}
indexedBStreeNode* c;//保存删除节点的孩子指针
if (p->leftchild != NULL) {
c = p->leftchild;
}
else {
c = p->rightchild;
}
//将孩子指针连接上
if (p == root) {
root = c;
}
else {
if (p == pp->leftchild) {
pp->leftchild = c;
}
else {
pp->rightchild = c;
}
}
size--;//树节点树减一
delete p;
cout << result << endl;
}
int main() {
int m;
scanf("%d", &m);
indexedBStree x;
for (int i = 0; i < m; i++) {
int a,b;
scanf("%d %d", &a, &b);
if (a == 0) {
x.insert(b);
}
else if (a == 1) {
x.find(b);
}
else if (a == 2) {
x.erase(b);
}
else if (a == 3) {
x.get(b);
}
else if (a == 4) {
x.idelete(b);
}
}
return 0;
}
如能打赏,不胜感激[叩谢]。