C++ -- 二叉搜索树

目录

1、二叉搜索树的概念

2、二叉搜索树的性能分析

3、二叉搜索树的实现

[3.1 中序遍历](#3.1 中序遍历)

[3.2 插入](#3.2 插入)

[3.3 查找](#3.3 查找)

[3.4 删除](#3.4 删除)

4、二叉搜索树key和key/value结构

[4.1 key搜索场景](#4.1 key搜索场景)

[4.2 key/value搜索场景](#4.2 key/value搜索场景)

[4.3 key_value结构的实现](#4.3 key_value结构的实现)

[4.4 拷贝构造和析构](#4.4 拷贝构造和析构)


由于 map 和 set 的底层是红黑树 ,包括后面要讲的 AVL 树 (平衡二叉搜索树),为了方便理解,我们先来讲解二叉搜索树,因为红黑树和 AVL 树都是在二叉搜索树的前提下实现的。

1、二叉搜索树的概念

二叉搜索树又称二叉排序树,它或者是一棵空树,或者是具有以下性质的二叉树:

  • 若它的左子树不为空,则左子树上所有结点的值都小于等于根结点的值
  • 若它的右子树不为空,则右子树上所有结点的值都大于等于根结点的值
  • 它的左右子树也分别为二叉搜索树

⼆叉搜索树中可以支持插入相等的值,也可以不支持插人相等的值,具体看使用场景定义,后续我们学习 map / set / multimap / multiset 系列容器底层就是二叉搜索树,其中 map / set 不之持插入相等值,multimap / multiset 支持插如相等值。

2、二叉搜索树的性能分析

二叉搜索树最多查找高度次。因此二叉搜索树的性能取决的树的高度

  • 最优情况下,二叉搜索树为完全⼆叉树(或者接近完全⼆叉树),其高度为: log2 N
  • 最差情况下,⼆叉搜索树退化为单⽀树(或者类似单⽀),其⾼度为: N

所以综合而言二叉搜索树增删查改时间复杂度为: O (N)

那么这样的效率显然是无法满足我们需求的,我们后续需要继续讲解⼆叉搜索树的变形,平衡二叉搜索树( AVL 树)和红黑树,才能适用于我们在内存中存储和搜索数据。

另外需要说明的是,二分查找 也可以实现 O (log2 N) 级别的查找效率,但是二分查找有两大缺陷:

  1. 需要存储在支持下标随机访问的结构中,并且有序。

  2. 插入和删除数据效率很低,因为存储在下标随机访问的结构中,插入和删除数据一般需要挪动数据。

这里也就体现出了平衡⼆叉搜索树的价值。

3、二叉搜索树的实现

首先我们要管理每个节点的数据以及它的左右孩子,所以我们先封装一个树节点类,用来管理节点的数据和左右孩子,其次我们要把每个树节点统一管理,所以还需要再封装一个树类,用来管理所有节点。

cpp 复制代码
template<class K>
struct BSTreeNode
{
	BSTreeNode(const K& key)
		:_key(key)
		,_left(nullptr)
		,_right(nullptr)
	{}
 
	K _key;
	BSTreeNode<K>* _left;
	BSTreeNode<K>* _right;
};
 
template<class K>
class BSTree
{
	//typedef BSTreeNode<K> Node;
	using Node = BSTreeNode<K>;
public:
private:
	Node* _root = nullptr;
};

3.1 中序遍历

因为 _root 是私有成员,所以我们可以这样多套一层调用。

cpp 复制代码
void Inorder()
{
	_Inorder(_root);
	cout << endl;
}
void _Inorder(const node* root)
{
	if (root == nullptr)
	{
		return;
	}
	_Inorder(root->left);
	cout << root->_key << " ";
	_Inorder(root->right);
}

3.2 插入

插入的具体过程如下:

  1. 树为空,则直接新增结点,赋值给 root 指针
  2. 树不空,按二叉搜索树性质,插入值比当前结点大往右走,插入值比当前结点小往左左,找到空位置,插入新结点。
  3. 如果支持插入相等的值,插入值跟当前结点相等的值可以往右走,也可以往左走,找到空位置,插入新结点。(要注意的是要保持逻辑⼀致性,插入相等的值不要⼀会往右走,⼀会往左走)
cpp 复制代码
bool Insert(const k& x)
{
	if (_root == nullptr)//插入根节点
	{
		_root = new node(x);
		return true;
	}
	node* cur = _root;
	node* parent = nullptr;//记录父亲节点
	while (cur)
	{
		//介质不冗余
		/*if (x < cur->_key)
		{
			parent = cur;
			cur = cur->left;
		}
		else if (x > cur->_key)
		{
			parent = cur;
			cur = cur->right;
		}
		else
		{
			return false;
		}*/
		//介质冗余
		if (x <= cur->_key)//相等插入左子树
		{
			parent = cur;
			cur = cur->left;
		}
		else if (x > cur->_key)
		{
			parent = cur;
			cur = cur->right;
		}
	}
	if (x > parent->_key)
	{
		parent->right = new node(x);
	}
	else//相等插入左子树
	{
		parent->left = new node(x);
	}
	return true;
}

3.3 查找

查找的具体过程如下:

  1. 从根开始比较,查找 x,x 比根的值大则往右边走查找,x 比根值小则往左边走查找。
  2. 最多查找高度次,走到空,还没找到,这个值不存在。
  3. 如果不支持插入相等的值,找到 x 即可返回
  4. 如果支持插入相等的值,意味着有多个 x 存在,一般要求查找中序的第一个 x。
cpp 复制代码
bool Find(const k& x)
{
	node* cur = _root;
	while (cur)
	{
		if (x < cur->_key)//节点在左子树
		{
			cur = cur->left;
		}
		else if (x > cur->_key)//节点在右子树
		{
			cur = cur->right;
		}
		else
		{
			return true;//找到target
		}
	}
	return false;
}

3.4 删除

首先查找元素是否在二叉搜索树中,如果不存在,则返回 false。

如果查找元素存在则分以下四种情况分别处理:(假设要删除的结点为 N)

  • 要删除结点 N 的左右孩子均为空
  • 要删除的结点 N 的左孩子为空,右孩子结点不为空
  • 要删除的结点 N 的右孩子为空,左孩子结点不为空
  • 要删除的结点 N 的左右孩子结点均不为空

如果删除根节点直接修改 _root,再删除节点即可。


cpp 复制代码
bool Erase(const k& x)
{
	node* cur = _root;
	node* parent = nullptr;
	while (cur)
	{
		if (x < cur->_key)
		{
			parent = cur;
			cur = cur->left;
		}
		else if (x > cur->_key)
		{
			parent = cur;
			cur = cur->right;
		}
		else//找到删除节点
		{
			if (cur->left == nullptr)//左为空
			{
				if (cur == _root)//防止删除根节点
				{
					_root = cur->right;
				}
				else//判断连接父亲的左子树还是右子树
				{
					if (cur == parent->left)
					{
						parent->left = cur->right;
					}
					else
					{
						parent->right = cur->right;
					}
				}
				delete cur;
				return true;
			}
			else if (cur->right == nullptr)
			{
				if (cur == _root)//防止删除根节点
				{
					_root = cur->left;
				}
				else//判断连接父亲的左子树还是右子树
				{
					if (cur == parent->left)
					{
						parent->left = cur->left;
					}
					else
					{
						parent->right = cur->left;
					}
				}
				delete cur;
				return true;
			}
			else
			{
			 // 策略:找右子树的最小节点(最左节点)作为替代节点
                node* replaceparent = cur;  // 替代节点的父节点,初始化为待删除节点,防止进不去循环
                node* replace = cur->right; // 替代节点初始化为待删除节点的右孩子
				while (replace->left)//寻找右子树的最小节点
				{
					replaceparent = replace;
					replace = replace->left;
				}
				cur->_key = replace->_key;// 用替代节点的键值覆盖待删除节点的键值
				// 处理替代节点的连接:替代节点只有右子树(或为空)
                // 判断替代节点是其父节点的左/右孩子
				if (replace == replaceparent->left)
				{
					replaceparent->left = replace->right;
				}
				else
				{
					replaceparent->right = replace->right;
				}
				delete replace;
				return true;
			}
		}
	}
	return false;
}

4、二叉搜索树key和key/value结构

4.1 key搜索场景

只有 key 作为关键码,结构中只需要存储 key 即可,关键码即为需要搜索到的值,搜索场景只需要判断 key ​​​​​​​在不在。key 的搜索场景实现的二叉树搜索树支持增删查,但是不支持修改,修改 key 破坏搜索树结构了。

  • 场景1:小区无人值守车库,小区车库买了车位的业主车才能进小区,那么物业会把买了车位的业主的车牌号录入后台系统,车辆进入时扫描车牌在不在系统中,在则抬杆,不在则提示非本小区车辆,无法进入。
  • 场景2:检查一篇英文章单词拼写是否正确,将词库中所有单词放入二叉搜索树,读取文章中的单 词,查找是否在二叉搜索树中,不在则波浪线标红提示

4.2 key/value搜索场景

每一个关键码 key,都有与之对应的值 value,即 <key, value> ,value 可以是任意类型对象。树的结构中(结点)除了需要存储 key 还要存储对应的 value,增/删/查还是以 key ​​​​​​​为关键字走二叉搜索树的规则进行比较,可以快速查找到 key ​​​​​​​对应的 value。key/value 的搜索场景实现的二叉树搜索树支持修改,但是不支持修改 key,修改 key 破坏搜索树性质了,可以修改 value。

  • 场景1:简单中英互译字典,树的结构中(结点)存储 key (英⽂)和 vlaue (中文),搜索时输入英文,则同时查找到了英文对应的中文。
  • 场景2:商场无人值守车库,入口进场时扫描车牌,记录车牌和入场时间,出口离场时,扫描车牌,查找入场时间,⽤ 当前时间 - 入场时间 计算出停车时长,计算出停车费用,缴费后抬杆,车辆离场。
  • 场景3:统计⼀篇文章中单词出现的次数,读取⼀个单词,查找单词是否存在,不存在这个说明第⼀次出现,(单词,1),单词存在,则 ++单词对应的次数。

4.3 key_value结构的实现

结构的定义:

cpp 复制代码
//key_value结构
template<class k, class v>
struct BSTNode
{
	using node = BSTNode<k, v>;
	k _key;
	v _value;
	node* left;
	node* right;
	BSTNode(const k& key, const v& value)
		:_key(key)
		, _value(value)
		, left(nullptr)
		, right(nullptr)
	{}
};
template<class k, class v>
class BSTree
{
	using node = BSTNode<k, v>;
public:
private:
	node* _root = nullptr;
};

查找和删除:逻辑不需要改变。但是如果想知道查找的节点的 value 可以返回节点指针。同时如果允许介质冗余的话,那就需要查找中序的第一个,所以我们相同值时,先保留,继续去左子树查找即可。

插入:只需要多插入一个 value 值即可,逻辑不变。

cpp 复制代码
bool Insert(const k& x, const v& v)
{
	if (_root == nullptr)//插入根节点
	{
		_root = new node(x, v);
		return true;
	}
	node* cur = _root;
	node* parent = nullptr;//保留父亲节点
	while (cur)
	{
		/*介质不冗余*/
		if (x < cur->_key)
		{
			parent = cur;
			cur = cur->left;
		}
		else if (x > cur->_key)
		{
			parent = cur;
			cur = cur->right;
		}
		else
		{
			return false;
		}
		//介质冗余
		//if (x <= cur->_key)//相等插入到左子树
		//{
		//	parent = cur;
		//	cur = cur->left;
		//}
		//else if (x > cur->_key)
		//{
		//	parent = cur;
		//	cur = cur->right;
		//}
	}
	if (x > parent->_key)
	{
		parent->right = new node(x, v);
	}
	else//相等插入左子树
	{
		parent->left = new node(x, v);
	}
	return true;
}
		

4.4 拷贝构造和析构

我们这里还要显式实现一下拷贝构造和析构,因为这里需要深拷贝资源,且在析构时要把每个 new 出来的树节点都要释放掉。

拷贝构造就直接 new 构造根节点,再让根节点连接递归构造的左右子树即可。注意如果按照插入的思路构造,那就必须是层序遍历节点插入,否则拷贝出来的子树就不一样。

cpp 复制代码
BSTree(const BSTree<k,v>& x)
{
	_root = Copy(x._root);
}
node* Copy( node* x)
{
	if (x == nullptr)
	{
		return x;
	}
	node* root = new node(x->_key, x->_value);
	root->left =Copy(x->left);
	root->right = Copy(x->right);
	return root;
}

析构:这里先析构左右子树在析构根节点即可。

cpp 复制代码
	void Destory(const node* root)
	{
		if (root == nullptr)
		{
			return ;
		}
		Destory(root->left);
		Destory(root->right);
		delete root;
	}
	~BSTree()
	{
		Destory(_root);
		_root = nullptr;
	}

赋值重载:这里先拷贝 tmp,然后交换 _root 指针,函数结束后析构 tmp 空间。

cpp 复制代码
BSTree<k, v>& operator = (BSTree<k, v> tmp)
{
	swap(_root,tmp._root);
	return *this;
}
相关推荐
TOYOAUTOMATON1 小时前
自动化工业夹爪
大数据·人工智能·算法·目标检测·机器人
im_AMBER1 小时前
Leetcode 67 长度为 K 子数组中的最大和 | 可获得的最大点数
数据结构·笔记·学习·算法·leetcode
charlie1145141911 小时前
深入理解CC++的编译与链接技术8:Windows和Linux是如何搜寻动态库的?
c语言·c++·动态库·编译·编译技术
郝学胜-神的一滴1 小时前
Linux信号四要素详解:从理论到实践
linux·服务器·开发语言·网络·c++·程序人生
yangpipi-1 小时前
《C++并发编程实战》 第3章 在线程间共享数据
开发语言·c++
fish_xk1 小时前
c++基础
开发语言·c++
互亿无线明明2 小时前
如何为全球业务构建可扩展的“群发国际短信接口”?
java·c++·python·golang·eclipse·php·erlang
buyue__2 小时前
C++实现数据结构——链表
数据结构·c++·链表
feifeigo1232 小时前
MATLAB实现两组点云ICP配准
开发语言·算法·matlab