【STL】二叉搜索树的讲解与全面实现

1.二叉搜索树的概念

二叉搜索树(BST,Binary Search Tree)又称二叉排序树,是一种特殊的二叉树,它或者是一棵空树,或者是具有以下性质的二叉树:

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

二叉搜索树(Binary Search Tree)的每个节点的左子树中的所有节点都比该节点小,右子树中的所有节点都比该节点大。这个特性使得二叉搜索树可以用来实现非常高效的查找、插入和删除操作。

2.二叉搜索树的功能

首先,在二叉搜索树的操作中只支持插入,查找,删除,遍历,并不支持修改操作,因为在修改后谁也不能保证它依然是一棵二叉搜索树,二叉搜索树的时间复杂度范围在(O(logN)~O(N))

在二叉搜索树的遍历中一般采用中序遍历: 先遍历左子树,然后访问根节点,最后遍历右子树。在BST中,中序遍历会按照升序访问所有节点

int a[] = {8, 3, 1, 10, 6, 4, 7, 14, 13};

3.二叉搜索树的实现

二叉搜索树结构的和树形结构差不多,这意味着每个元素(通常称为节点)都有两个指针:一个指向前一个左子树,另一个指向右子树,因此我们需要单独再定义一个类来表示节点结构,每个节点再串联起来构成BST

3.1  节点的定义

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

3.2 . BST的定义

template<class K>
class BSTree
{
	typedef BSTreeNode<K> Node;
public:
	// 构造函数等可能的其他成员函数... 
private:
	Node* _root = nullptr;
};

3.3 二叉搜索树的操作

1.查找(find)

a、从根开始比较,查找,比根大则往右边走查找,比根小则往左边走查找。
b、最多查找高度次,走到到空,还没找到,这个值不存在。

bool Find(const K& key) 
{
	Node* cur = _root;
	while (cur) {
		if (cur->_key < key) { //查找的值比cur小,往左走
			cur = cur->_right;
		}
		else if (cur->_key > key) { //查找的值比cur大,往右走
			cur = cur->_left;
		}
		else  return true;//找到就返回true
	}
	return false; //找不到
}

2.插入(insert)

a、树为空,则直接新增节点,赋值给root指针
b、树不空,按二叉搜索树性质查找插入位置,插入新节点

bool Insert(const K&key) //需要找个节点保留插入节点的父节点,先找到要插入位置
{
	if (_root == nullptr) { //根为空时直接插入
		_root = new Node(key); 
		return true;
	}
	// 定义parent是因为,在最后找到插入位置时,需要parent将节点进行连接
	Node* parent = nullptr;
	Node* cur = _root;
	while (cur) {
		if (cur->_key < key) {// 插入的值比cur位置大,cur往右走
			parent = cur;
			cur = cur->_right;
		}
		else if (cur->_key > key) {// 插入的值比cur位置小,cur往左走
			parent = cur;
			cur = cur->_left;
		}
        // 当插入的值和cur位置相等时,直接退出,因为二叉搜索树不允许有相同的元素
		else  return false;
	}
	//新插入节点与二叉搜索树连接
	cur = new Node(key);
	if (parent->_key < key) parent->_right = cur;
	else parent->_left = cur;
	
	return true;
}

3.删除(erase)

首先查找元素是否在二叉搜索树中,如果不存在,则返回, 否则要删除的结点可能分下面四种情况:
a. 要删除的结点无孩子结点
b. 要删除的结点只有左孩子结点
c. 要删除的结点只有右孩子结点
d. 要删除的结点有左、右孩子结点


看起来有待删除节点有4中情况,但是实际情况a可以与情况b或者c合并起来,因此真正的删

除过程如下:
 

情况b:删除该结点且使被删除节点的双亲结点指向被删除节点的左孩子结点--直接删除
情况c:删除该结点且使被删除节点的双亲结点指向被删除结点的右孩子结点--直接删除
情况d:在它的右子树中寻找中序下的第一个结点(关键码最小),用它的值填补到被删除节点中,再来处理该结点的删除问题--替换法删除

// 从二叉搜索树中删除指定的键值对应的节点
bool Erase(const K& key)
{
    Node* parent = nullptr; // 父节点指针,初始化为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 (parent == nullptr)
                    _root = cur->_right; // 将根节点指向当前节点的右子节点
                else 
                {
                    // 根据当前节点在父节点中的位置,更新父节点的相应子节点指针
                    if (parent->_left == cur)
                        parent->_left = cur->_right;
                    else
                        parent->_right = cur->_right;
                }

                delete cur; // 删除当前节点
                return true;
            }
            // 如果当前节点的右子节点为空,即当前节点最多有一个左子节点
            else if (cur->_right == nullptr) 
            {
                // 如果没有父节点,则当前节点是根节点
                if (parent == nullptr)
                    _root = cur->_left; // 将根节点指向当前节点的左子节点
                else 
                {
                    // 根据当前节点在父节点中的位置,更新父节点的相应子节点指针
                    if (parent->_left == cur)
                        parent->_left = cur->_left;
                    else
                        parent->_right = cur->_left;
                }

                delete cur; // 删除当前节点
                return true;
            }
            else 
            { // 当前节点有两个子节点
                // 找到当前节点右子树中的最小节点(即右子树的最左节点)
                Node* rightMinParent = cur;
                Node* rightMin = cur->_right;
                while (rightMin->_left) 
                {
                    rightMinParent = rightMin;
                    rightMin = rightMin->_left;
                }
                // 用右子树中最小节点的值替换当前节点的值
                cur->_key = rightMin->_key;

                // 从右子树中删除最小节点
                if (rightMinParent->_left == rightMin) // 如果 rightMin 是 rightMinParent 的左子节点
                    rightMinParent->_left = rightMin->_right; // 更新 rightMinParent 的左子节点指针为 rightMin 的右子节点
                else // 如果 rightMin 是 rightMinParent 的右子节点
                    rightMinParent->_right = rightMin->_right; // 更新 rightMinParent 的右子节点指针为 rightMin 的右子节点

                delete rightMin; // 删除最小节点
                return true;
            }
        }
    }
    return false; // 如果没有找到要删除的节点
}

4.搜索树的遍历(中序遍历)

	void Inorder() //避免_root私有,无法提供的问题
	{
		_Inorder(_root);
	}
private:
	void _Inorder(Node* root)
	{
		if (root == nullptr) return; //递归截止
		_Inorder(root->_left);
		printf("%d ", root->_key);
		_Inorder(root->_right);
	}

5.二叉搜索树默认成员函数

5.1构造

BSTree() = default; // 显式地定义默认构造函数  

5.2拷贝构造

BSTree(const BSTree<K>& t)
{
	_root = Copy(t._root);
}
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;
	}

5.3赋值重载

BSTree<K>& operator=(BSTree<K> t)
{
    // 现代写法-> 直接调用swap
    swap(_root, t._root);
    return *this;
}

5.4析构

~BSTree()
{
	Destory(_root);
}
private:
	void Destroy(Node* root)
    {
	    if (root == nullptr) return;
	    Destroy(root->_left);
	    Destroy(root->_right);
	    delete root;
    }

4.二叉搜索树的KV模型

KV模型:每一个关键码key,都有与之对应的值Value,即<Key, Value>的键值对。该种方式在现实生活中非常常见:

比如英汉词典就是英文与中文的对应关系,通过英文可以快速找到与其对应的中文,英
文单词与其对应的中文<word, chinese>就构成一种键值对
再比如统计单词次数,统计成功后,给定单词就可快速找到其出现的次数,单词与其出
现次数就是<word, count>就构成一种键值对

和上面二叉搜索树建立类似,在成员函数中,我们只需要在Insert中加入value元素即可,

如: BSTNode<K> Node 变成  BSTNode<K, V> Node , 

如new Node(key); 变成  new Node(key, value);  

还有就是find不再是返回bool值,而是返回的节点值

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.2KV模型的实现

namespace keyValue  // 避免与之前k模型冲突
{
	template<class K, class V>
	struct BSTreeNode
	{
		BSTreeNode<K>* _left;
		BSTreeNode<K>* _right;
		K _key;
		V _value;
		
		BSTreeNode(const K& key = K(), const V& value = V())
			: _left(nullptr)
			, _right(nullptr)
			, _key(key)
			, _value(value)
		{}
	};
	template<class K, class V>
	class BSTree
	{
		typedef BSTreeNode<K, V> Node;
	public:
		// 构造函数等可能的其他成员函数... 
		// 在成员函数中,我们只需要在insert中加入value元素即可
	private:
		Node* _root = nullptr;
	};
}

代码测试

void test2()
{
	// 输入单词,查找单词对应的中文翻译
	keyValue::BSTree<string, string> dict;
	dict.Insert("string", "字符串");
	dict.Insert("tree", "树");
	dict.Insert("left", "左边、剩余");
	dict.Insert("right", "右边");
	dict.Insert("sort", "排序");
	// 插入词库中所有单词
	string str;
	while (cin >> str)
	{
		auto ret = dict.Find(str);
		if (ret == nullptr)
		{
			cout << "单词拼写错误,词库中没有这个单词:" << str << endl;
		}
		else
		{
			cout << str << "中文翻译:" << ret->_value << endl;
		}
	}
}

2.计数

void test3()
{
	// 统计水果出现的次数
	string arr[] = { "苹果", "西瓜", "苹果", "西瓜", "苹果", "苹果", "西瓜",
	"苹果", "香蕉", "苹果", "香蕉" };
	keyValue::BSTree<string, int> countTree;
	for (const auto& str : arr)
	{
		// 先查找水果在不在搜索树中
		// 1、不在,说明水果第一次出现,则插入<水果, 1>
		// 2、在,则查找到的节点中水果对应的次数++
		//BSTreeNode<string, int>* ret = countTree.Find(str);
		auto ret = countTree.Find(str);
		if (ret == NULL)
		{
			countTree.Insert(str, 1);
		}
		else
		{
			ret->_value++;
		}
	}
	countTree.InOrder();
}

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

插入和删除操作都必须先查找,查找效率代表了二叉搜索树中各个操作的性能。
  对有n个结点的二叉搜索树,若每个元素查找的概率相等,则二叉搜索树平均查找长度是结点在二叉搜索树的深度的函数,即结点越深,则比较次数越多。
  但对于同一个关键码集合,如果各关键码插入的次序不同,可能得到不同结构的二叉搜索树:

最优情况下,二叉搜索树为完全二叉树(或者接近完全二叉树),其平均比较次数为:\log N

最差情况下,二叉搜索树退化为单支树(或者类似单支),其平均比较次数为:\frac{N}{2}
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

北海有初拥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值