目錄
前言
1.二叉搜索樹的概念
1.1基本結構
1.2性能分析
2.二叉搜索樹的實現
2.1創建
2.2插入
2.3查找與遍歷
2.4刪除
3.二叉搜索樹類代碼
前言
C++中STL的map與set容器廣泛應用于實踐過程中,本文將詳細分析容器最基礎的二叉搜索樹結構,為后續map與set容器的學習打下更好基礎。更多C++內容可前往->|C++主題專欄|<-
1.二叉搜索樹的概念
1.1基本結構
二叉搜索樹是一種便于快速查找(搜索)數據的搜索結構,也稱二叉排序樹。它或是棵空樹,又或是滿足以下性質的樹:
二叉搜索樹的性質
? 若它的左?樹不為空,則左?樹上所有結點的值都?于等于根結點的值
? 若它的右?樹不為空,則右?樹上所有結點的值都?于等于根結點的值
? 它的左右?樹也分別為?叉搜索樹
? ?叉搜索樹中可以?持插?相等的值,也可以不?持插?相等的值,具體看使?場景定義
如上兩棵樹都是二叉搜索樹,當然第二棵樹的重復元素(10)可以在父節點的左子樹也可以在父節點的右子樹。但我們常用的一般是左邊無重復元素的二叉搜索樹。
1.2性能分析
以上面第一棵樹為例:這種情況下的樹基本為完全二叉樹,其搜索性能(時間復雜度)與樹的高度相同為logN。但這是最優的情況,若二叉搜索樹的數據插入滿足一定條件,就會形成類似單支樹的情況(如下圖),這種情況下的搜索性能就會降為O(N)。
因此綜合考慮,普通搜索二叉樹的搜索性能就是O(N),但這樣的效率顯然不夠理想,后面在實際應用時使用的是優化過平衡二叉樹(AVL樹、紅黑樹)。
其次對于查找功能,在數組中的二分查找算法也能實現O(logN)級別的查找效率,但二分查找算法對比于二叉搜索樹有著明顯的缺點:
二分查找的缺點
1. 需要存儲在?持下標隨機訪問的結構中,并且有序
2. 插?和刪除數據效率很低。存儲在下標隨機訪問的結構中,插?和刪除數據?般需要挪動數據
這里二叉搜索樹的價值也更為顯著。
2.二叉搜索樹的實現
二叉搜索樹的核心功能是增、刪、查功能的實現,修改數據值的話主要會破壞搜索結構,進而大大降低效率,因此一般不會實現。下面會從邏輯->代碼依次實現各部分功能。
2.1創建
二叉搜索的底層結構還是一個鏈式二叉樹,因此需要結點,主體部分用根來連接各個結點即可。結點中需要儲存數據,以及左右孩子結點。但要注意的是搜索結構中顯示進行搜索的數據我們一般稱為關鍵字(key)。
其次結點的構造函數只需保證讓key值的構造,左右孩子指針置空即可。代碼如下:
#pragma once
#include<iostream>
using namespace std;//樹結點
template<class K>
struct BSTNode
{K _key;BSTNode<K>* _left;BSTNode<K>* _right;BSTNode(K x):_key(x),_left(nullptr),_right(nullptr){}
};//二叉搜索樹
template<class K>
class BSTree
{
private:typedef BSTNode<K> Node
public:private:Node* _root = nullptr;
};
2.2插入
插入分為為空插入與非空插入。當樹為空的時候只需讓根結點指向新結點即可。
當樹非空時,按照二叉搜索樹的性質,插入值比當前結點大就往右子樹走,插入值比當前結點小就往左子樹走。
當遇見插入值與當前結點相同時就可以直接返回,表示結構中已有此值。但為了區分void返回的插入成功與失敗,我們插入函數的返回類似設置為bool。
以在下圖樹中插入5為例:
????????????????????????????????????????
????????
這里最需要注意的是最后一部得到5應該插入的位置時,那個位置是為空的,因此需要在過程中記錄應該插入位置的父結點。代碼如下:
//插入
bool insert(const K& key)
{//分為根為空或不為空if (_root == nullptr){_root = new Node(key);return true;}//不為空則尋找合適位置Node* cur = _root;Node* parent = nullptr;while (cur){parent = cur;if (key > cur->_key){cur = cur->_right;}else if (key < cur->_key){cur = cur->_left;}else if(key == cur->_key){return false; //不插入重復數據}}//插入左還是右if (key > parent->_key){parent->_right = new Node(key);}else{parent->_left = new Node(key);}return true;
}
除了迭代的方式插入,也可通過遞歸進行插入:
private://遞歸插入bool _insert(Node*& root, const K& x){if (root == nullptr){root = new Node(x);return true;}if (root->_key == x){return false;}return root->_key > x ? _insert(root->_left, x) : _insert(root->_right, x);}public://遞歸插入bool REinsert(const K& x){return _insert(_root, x);}
這里最值得注意的是遞歸插入利用引用特性,能直接修改最后空位置的結點。
2.3查找與遍歷
二叉搜索樹的查找邏輯與插入類似,不同的是遇見相同的值就返回當前結點,若樹遍歷完沒找到就返回空。但查找的前提是樹不為空。
二叉搜索樹的遍歷主要實現方式為中序遍歷,這部分主要注意的是在類中顯示成員函數實現遞歸,后面代碼實現后再進行分析。
private://方便函數內部傳參void _inorder(Node* root){if (root == nullptr){return;}_inorder(root->_left);cout << root->_key << ' ';_inorder(root->_right);}
public://遍歷——中序遍歷void inorder(){_inorder(_root);cout << endl;}//查找Node* find(const K& x){assert(_root != nullptr);Node* cur = _root;while (cur){if (cur->_key < x){cur = cur->_right;}else if (cur->_key > x){cur = cur->_left;}else if(cur->_key == x){return cur;}}return nullptr;}
?遞歸調用時因要傳參數,而在類外部無法直接傳樹的根結點,因此可以直接在類中進行內部函數嵌套,進而實現遞歸調用。
2.4刪除
二叉搜索樹的刪除是最復雜的,因為當結點刪除時可能也會涉及搜索結構的破壞,但這里的破壞情況會比修改簡單。
具體刪除情況可以分為以下三種(均以下圖樹為例):
???????????????????????????????
1.要刪除結點N左右孩?均為空。這時可以直接將改結點刪除,并將父節點置為空。
????????
2.要刪除的結點N左孩?或右孩子為空,另一孩子結點不為空。這種情況下要刪改結點,可以直接讓該結點的父親指向該結點的孩子結點。
3.要刪除的結點N左右孩?結點均不為空。這種情況?法直接刪除N結點,因為N的兩個孩??處安放,只能?替換法刪除。找N左?樹的值最?結點R(最右結點)或者N右?樹的值最?結點R(最左結點)替代N,因為這兩個結點中任意?個,放到N的位置,都滿??叉搜索樹的規則。替代N的意思就是N和R的兩個結點的值交換,轉?變成刪除R結點,R結點符合情況2或情況3,可以直接刪除。
可以看到這里本質是讓其變為第一種情況再進行刪除。
綜上進行代碼實現:
public://刪除bool erase(const K& x){//內部查找Node* cur = _root;Node* parent = nullptr;while (cur){parent = cur;if (cur->_key < x){cur = cur->_right;}else if (cur->_key > x){cur = cur->_left;}else //找到了,進行分情況刪除{if(cur->_left == nullptr) //左孩子為空{if (cur != _root){//父親指向不為空的右if (cur == parent->_left){parent->_left = cur->_right;}else{parent->_right = cur->_right;}}else{_root = cur->_right;}delete cur;}else if(cur->_right == nullptr) //右孩子為空{if (cur != _root){//父親指向不為空的右if (cur == parent->_left){parent->_left = cur->_left;}else{parent->_right = cur->_left;}}else{_root = cur->_left;}delete cur;}else //左右都不為空{//查找合適結點——左子樹最大Node* curLeftmax = cur->_left;Node* maxParent = cur;while (curLeftmax->_right){maxParent = curLeftmax;curLeftmax = curLeftmax->_right;}//交換key值swap(cur->_key, curLeftmax->_key);if (curLeftmax == maxParent->_left)maxParent->_left = curLeftmax->_right;elsemaxParent->_right = curLeftmax->_right;delete curLeftmax;}return true;}}return false;}
這里實際情況并沒有將第一種分化出來,因為當被刪除結點的孩子都為空時,其父親無論指向左孩子還是右孩子仍然同樣置空。
其次是最后一種情況只交換兩個結點的值,不用再進行復雜的結點遷移,利于直接保持原樹順序。
3.二叉搜索樹類代碼
#pragma once
#include<iostream>
#include<assert.h>
#include<algorithm>
using namespace std;//樹結點
template<class K>
struct BSTNode
{K _key;BSTNode<K>* _left;BSTNode<K>* _right;BSTNode(K x):_key(x),_left(nullptr),_right(nullptr){}
};//二叉搜索樹
template<class K>
class BSTree
{
private:typedef BSTNode<K> Node;//方便函數內部傳參void _inorder(Node* root){if (root == nullptr){return;}_inorder(root->_left);cout << root->_key << ' ';_inorder(root->_right);}//遞歸插入bool _insert(Node*& root, const K& x){if (root == nullptr){root = new Node(x);return true;}if (root->_key == x){return false;}return root->_key > x ? _insert(root->_left, x) : _insert(root->_right, x);}public:BSTree() = default;//調用默認構造即可//插入bool insert(const K& key){//分為根為空或不為空if (_root == nullptr){_root = new Node(key);return true;}//不為空則尋找合適位置Node* cur = _root;Node* parent = nullptr;while (cur){parent = cur;if (key > cur->_key){cur = cur->_right;}else if (key < cur->_key){cur = cur->_left;}else if(key == cur->_key){return false; //不插入重復數據}}if (key > parent->_key){parent->_right = new Node(key);}else{parent->_left = new Node(key);}return true;}//遞歸插入bool REinsert(const K& x){return _insert(_root, x);}//刪除bool erase(const K& x){//內部查找Node* cur = _root;Node* parent = nullptr;while (cur){parent = cur;if (cur->_key < x){cur = cur->_right;}else if (cur->_key > x){cur = cur->_left;}else //找到了,進行分情況刪除{if(cur->_left == nullptr) //左孩子為空{if (cur != _root){//父親指向不為空的右if (cur == parent->_left){parent->_left = cur->_right;}else{parent->_right = cur->_right;}}else{_root = cur->_right;}delete cur;}else if(cur->_right == nullptr) //右孩子為空{if (cur != _root){//父親指向不為空的右if (cur == parent->_left){parent->_left = cur->_left;}else{parent->_right = cur->_left;}}else{_root = cur->_left;}delete cur;}else //左右都不為空{//查找合適結點——左子樹最大Node* curLeftmax = cur->_left;Node* maxParent = cur;while (curLeftmax->_right){maxParent = curLeftmax;curLeftmax = curLeftmax->_right;}//交換key值swap(cur->_key, curLeftmax->_key);if (curLeftmax == maxParent->_left)maxParent->_left = curLeftmax->_right;elsemaxParent->_right = curLeftmax->_right;delete curLeftmax;}return true;}}return false;}//遍歷——中序遍歷void inorder(){_inorder(_root);cout << endl;}//查找Node* find(const K& x){assert(_root != nullptr);Node* cur = _root;while (cur){if (cur->_key < x){cur = cur->_right;}else if (cur->_key > x){cur = cur->_left;}else if(cur->_key == x){return cur;}}return nullptr;}private:Node* _root = nullptr;
};