[数据结构]二叉搜索树

今天我们来学习一下新的数据结构,二叉搜索树,这个结构比较简单,是一个铺垫式的结构,为之后的平衡二叉树,AVL树以及红黑树做一个知识基础,我们将从概念到实现具体的介绍二叉搜索树。

目录

Ⅰ.二叉搜索树的概念

Ⅱ.二叉搜索树的性能分析

Ⅲ.二叉搜索树的具体实现

1.二叉搜索树的节点

2.二叉搜索树的插入

3.二叉搜索树的查找

4.二叉搜索树的删除

5.二叉搜索树的销毁

6.二叉搜索树的遍历

7.二叉搜索树的拷贝

Ⅳ.二叉搜索树key和key/value


Ⅰ.二叉搜索树的概念

我们之前已经学习过二叉树的结构了,从名字我们就可以很容易的看出来,二叉搜索树也是二叉树的一种,他的大体结构还是和二叉树一致,只是有一些逻辑上的差异。

我们先看一下二叉搜索树的模型图

我们通过上面的图可以得到几个结论

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

  • 若他的左子树不为空,则左子树上面的所有值都小于根节点的值
  • 若他的右子树不为空,则右子树上面的所有值都大于根节点的值
  • 它的左右子树也都是二叉搜索树
  • 二叉搜索树中可以支持插入相等的值,也可以不支持插入相等的值,具体看使用场景定义

Ⅱ.二叉搜索树的性能分析

我们可以先看一下,二叉搜索树的最优结构和最坏结构如下所示

最优情况下,二叉搜索树为完全二叉树(或者接近完全二叉树),其高度为: O(log2 N)

最差情况下,二叉搜索树退化为单支树(或者类似单支),其高度为: O(N)

综合下来,二叉搜索树的增删查的时间复杂度为O(N)

这样看二叉搜索树的增删改的时间复杂度与数组一样,那这样的效率就属实有点过低了,所以在后面我们将以此基础实现平衡二叉树等效率高的结构,二叉搜索树只是一个基础。

Ⅲ.二叉搜索树的具体实现

1.二叉搜索树的节点

既然是树,那么依然需要树的根节点结构,即_left——左子树,_right——右子树,_key——自身节点,具体代码实现如下

template<class K>
struct BSTNode
{
K _key;
BSTNode<K>* _left;
BSTNode<K>* _right;
BSTNode(const K& key)
:_key(key)
, _left(nullptr)
, _right(nullptr)
{}
};

2.二叉搜索树的插入

我们将二叉搜索树的增删查都放到另一个类中,此类中含有唯一私有成员_root

插入分为二种情况

  1. 树为空,则直接新增结点,赋值给root指针
  2. 树不空,按二叉搜索树性质,插入值比当前结点大往右走,插入值比当前结点小往左走,找到空位置,插入新结点

我们可以控制是否插入相同的值,但是注意,插入相同的值时,需要保持逻辑一致,不能一会向左子树插,一会向右子树插入

逻辑:

我们这里以不插入相同元素为例

先判断根节点是否为空,若为空就直接新增节点,给root赋值

若不为空,则需要创建两个指针变量,一个cur指向根节点,另一个parent指向cur的父节点,来确定树的结构

我们创建一个while循环,当cur为空时停止循环,先让parent赋值为cur,然后用分支语句进行判断,若传入的值key大于cur的_key就让cur向右走,反之向左走,若相等就返回false中止插入。

当cur等于空时,则找到了正确的位置,此时只需要新建一个节点,并再次通过分支语句,判断该节点是parent的左子树还是右子树,连接即可,返回true,表示插入成功。

bool Insert(const K& key)
{
	if (_root == nullptr)
	{
		_root = new Node(key);
		return true;
	}
	Node* parent = nullptr;
	Node* cur = _root;

	while (cur)
	{
		if (cur->_key < key)
		{
			parent = cur;
			cur = cur->_right;
		}
		else if (cur->_key > key)
		{
			parent = cur;
			cur = cur->_left;
		}
		else
			return false;
	}
	cur = new Node(key);
	if (parent->_key < key)
	{
		parent->_right = cur;
	}
	else
	{
		parent->_left = cur;

	}
	return true;

}

3.二叉搜索树的查找

二叉搜索树的查找连接和插入逻辑类似,但是更为简单,不需要parent节点,在按照分支逻辑走即可

基本逻辑

1.从根开始比较,查找key,key比根的值大则往右边走查找,key比根值小则往左边走查找。
2. 最多查找高度次,走到到空,还没找到,这个值不存在。
3. 如果不支持插入相等的值,找到key即可返回
4. 如果支持插入相等的值,意味着有多个key存在,⼀般要求查找中序的第⼀个key。

我们看一下插入的图

代码实现

		Node* Find(const K & key)
		{
			Node* cur = _root;
			while (cur)
			{
				if (cur->_key < key)
				{
					cur = cur->_right;
				}
				else if (cur->_key > key)
				{
					cur = cur->_left;
				}
				else
					return cur;
			}
			return nullptr;
		}

4.二叉搜索树的删除

二叉搜索树的删除,是二叉搜索树实现最为复杂的部分

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

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

  1.  要删除结点N左右孩子均为空
  2. 要删除的结点N左孩子位空,右孩子结点不为空
  3. 要删除的结点N右孩子位空,左孩子结点不为空
  4. 要删除的结点N左右孩子结点均不为空

由于前三种情况的解决方式是一致的,可以归类为同一种情况

第一种 把N结点的父亲对应孩子指针指向空,直接删除N节点

第二种 把N节点的父亲对应孩子的指针指向N的右孩子,直接删除N节点

第三种 把N节点的父亲对应孩子的指针指向N的左孩子,直接删除N节点

这个便是以上三种情况的例子,以及解决方案

第四种情况,就比较特殊了,由于其具有两个孩子节点,而对应的父节点已经有两个子节点了,即使删掉一个子节点,那么还缺一个子节点的位置,那么该如何解决呢?

这里我们将运用替换节点的思路,来解决这个情况的问题

找N左子树的值最大结点R(最右结点)或者N右子树的值最小结点R(最左结点)替代N,因为这两个结点中任意一个,放到N的位置,都满足二叉搜索树的规则。替代N的意思就是N和R的两个结点的值交换,转而变成删除R结点,R结点符合情况2或情况3,可以直接删除

如图所示

我们在删除部分依旧可以复用查找和插入的代码,只需要在比较cur的成员_key和传入变量key相等的部分,来实现我们的删除逻辑即可,代码实现如下所示

bool Erase(const K & key)
{
	Node* parent = nullptr;
	Node* cur = _root;
	while (cur)
	{
		if (cur->_key < key)
		{
			parent = cur;
			cur = cur->_right;
		}
		else if (cur->_key > key)
		{
			parent = cur;
			cur = cur->left;
		}
		else
		{
			if (cur->_left == nullptr)
			{
				if (cur == _root)
				{
					_root = cur ->_right;
				}
				else
				{
					if (parent->_left == cur)
					{
						parent->_left = cur->_right;
					}
					else
					{
						parent->_right = cur->_right;
					}
				}
				delete cur;
			}
			else if (cur->_right == nullptr)
			{
				if (cur == _root)
				{
					_root = cur->_left;
				}
				else
				{
					if (parent->_left == cur)
					{
						parent->_left = cur->_left;
					}
					else
					{
						parent->_right = cur->_left;
					}
				}
				delete cur;
			}
			else
			{
				Node* replaceParent = cur;
				Node* replace = cur->_right;
				while (replace->_left)
				{
					replaceParent = replace;
					replace = replace->_left;
				}
				cur->_key = replace->_key;
				cur->_value = replace->_value;
				if (replaceParent->_left == replace)
					replace->_left = replace->_right;
				else
					replace->_right = replace->_right;
				delete replace;

			}
		}

	}
}

以上是二叉树的基本增删查的实现部分,可以看到这里并没有改的操作,因为二叉搜索树一旦修改某值,整个树的结构就发生了改变,就不满足二叉搜索树的结构了。

所以基本的应用,二叉搜索树只有增删查功能

下面我们实现一些树的基本功能,这部分将会使用到递归的思想,如果忘了可以去复习一下二叉树的基本用法

5.二叉搜索树的销毁

二叉搜索树的销毁,我们就只需要按照后序遍历的顺序依次对节点进行释放即可。

代码实现如下

	void Destroy(Node* root)
	{
		if (root == nullptr)
			return;
		Destroy(root->_left);
		Destroy(root->_right);
		delete root;
	}

6.二叉搜索树的遍历

二叉搜索树的中序遍历即将二叉树按从小到大的顺序进行打印,所以我们只需要实现中序遍历即可

运行图

代码如下

		void _InOrder(Node* root)
		{
			if (root == nullptr)
				return;
			_InOrder(root->_left);
			cout << root->_key << " ";
			_InOrder(root->_right);

		}

7.二叉搜索树的拷贝

这个部分有些人会想着按照前序遍历一个一个插入即可,但是这样是万万不可的,会导致二叉搜索树的结构被破坏,如果必须想着用一个一个插入,那么可以用层序遍历的顺序进行插入,除此之外,我们可以用递归的思想进行拷贝,代码很容易,逻辑需要略加思考

代码如下 

		Node* Copy(Node* root)
		{
			if (root == nullptr)
				return nullptr;

			Node* newRoot = new Node(root->_key, root->_value);
			newRoot->_left = Copy(root->_left);
			newRoot->_right = Copy(root->_right);
			return newRoot;
		}

Ⅳ.二叉搜索树key和key/value

上面实现的就是二叉搜索树的key模型

我们二叉搜索树还可以实现key/value模型

每⼀个关键码key,都有与之对应的值value,value可以任意类型对象。树的结构中(结点)除了需要存储key还要存储对应的value,增/删/查还是以key为关键字走二叉搜索树的规则进行比较,可以快速查找到key对应的value。

整个代码如下所示,key模型和key/value模型的代码实现都是大差不差,只是key/value模型的模板会多出一个类型,能够存储value的值

整段代码实现如下

#include<iostream>
using namespace std;
namespace bit 
{
template<class K, class V>
struct BSTNode
{
	K _key;
	V _value;

	BSTNode<K, V>* _left;
	BSTNode<K, V>* _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:
	BSTree() = default;


	BSTree(const BSTree& t)
	{
		_root = Copy(t.root);
	}

	BSTree& operator=(BSTree tmp)
	{
		swap(_root, tmp._root);
		return *this;
	}

	~BSTree()
	{
		Destroy(_root);
		_root = nullptr;
	}

	bool Insert(const K& key, const V& value)
	{
		if (_root == nullptr)
		{
			_root = new Node(key, value);
			return true;
		}
		Node* parent = nullptr;
		Node* cur = _root;

		while (cur)
		{
			if (cur->_key < key)
			{
				parent = cur;
				cur = cur->_right;
			}
			else if (cur->_key > key)
			{
				parent = cur;
				cur = cur->_left;
			}
			else
				return false;
		}
		cur = new Node(key, value);
		if (parent->_key < key)
		{
			parent->_right = cur;
		}
		else
		{
			parent->_left = cur;

		}
		return true;

	}

		Node* Find(const K & key)
		{
			Node* cur = _root;
			while (cur)
			{
				if (cur->_key < key)
				{
					cur = cur->_right;
				}
				else if (cur->_key > key)
				{
					cur = cur->_left;
				}
				else
					return cur;
			}
			return nullptr;
		}

		bool Erase(const K & key)
		{
			Node* parent = nullptr;
			Node* cur = _root;
			while (cur)
			{
				if (cur->_key < key)
				{
					parent = cur;
					cur = cur->_right;
				}
				else if (cur->_key > key)
				{
					parent = cur;
					cur = cur->left;
				}
				else
				{
					if (cur->_left == nullptr)
					{
						if (cur == _root)
						{
							_root = cur ->_right;
						}
						else
						{
							if (parent->_left == cur)
							{
								parent->_left = cur->_right;
							}
							else
							{
								parent->_right = cur->_right;
							}
						}
						delete cur;
					}
					else if (cur->_right == nullptr)
					{
						if (cur == _root)
						{
							_root = cur->_left;
						}
						else
						{
							if (parent->_left == cur)
							{
								parent->_left = cur->_left;
							}
							else
							{
								parent->_right = cur->_left;
							}
						}
						delete cur;
					}
					else
					{
						Node* replaceParent = cur;
						Node* replace = cur->_right;
						while (replace->_left)
						{
							replaceParent = replace;
							replace = replace->_left;
						}
						cur->_key = replace->_key;
						cur->_value = replace->_value;
						if (replaceParent->_left == replace)
							replace->_left = replace->_right;
						else
							replace->_right = replace->_right;
						delete replace;

					}
				}

			}
		}

	

		void InOrder()
		{
			_InOrder(_root);
			cout << endl;
		}


	private:

		Node* Copy(Node* root)
		{
			if (root == nullptr)
				return nullptr;

			Node* newRoot = new Node(root->_key, root->_value);
			newRoot->_left = Copy(root->_left);
			newRoot->_right = Copy(root->_right);
			return newRoot;
		}

		void Destroy(Node* root)
		{
			if (root == nullptr)
				return;
			Destroy(root->_left);
			Destroy(root->_right);
			delete root;
		}

		void _InOrder(Node* root)
		{
			if (root == nullptr)
				return;
			_InOrder(root->_left);
			cout << root->_key << " ";
			_InOrder(root->_right);

		}

		Node* _root = nullptr;
	};
}

以上便是搜索二叉树的具体知识点与实现,希望对你带来帮助!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值