目录
搜索二叉树的概念:
二叉搜索树:一棵二叉树,可以为空;如果不为空,满足以下性质:
若它的左子树不为空,则左子树的值小于根结点
若它的右子树不为空,则右子树的值大于根结点
它的左右子树也分别是搜索二叉树
搜索二叉树的操作:
插入:
a. 树为空,则直接新增节点,赋值给root指针。b. 树不空,按二叉搜索树性质查找插入位置,插入新节点。
查找:
a. 从根开始比较,查找,比根大则往右边走查找,比根小则往左边走查找。b. 最多查找高度次,走到到空,还没找到,这个值不存在。
删除:
首先查找元素是否在二叉搜索树中,如果不存在,则返回 , 否则要删除的结点可能分下面四种情 况:a. 要删除的结点无孩子结点b. 要删除的结点只有左孩子结点c. 要删除的结点只有右孩子结点d. 要删除的结点有左、右孩子结点待删除节点有 4 中情况,实际情况 a 可以与情况 b 或者 c 合并起来,因此真正的删除过程如下: 情况b :删除该结点且使被删除节点的双亲结点指向被删除节点的左孩子结点 -- 直接删除情况c :删除该结点且使被删除节点的双亲结点指向被删除结点的右孩子结点 -- 直接删除情况d :在它的右子树中寻找中序下的第一个结点 ( 关键码最小 ),用它的值填补到被删除节点中,再来处理该结点的删除问题 -- 替换法删除
搜索二叉树的实现:
Key模型:
namespace key
{
template <class T>
struct BSTNode
{
T _data;
BSTNode<T>* _left;
BSTNode<T>* _right;
BSTNode(const T& data = T())
:_left(nullptr)
, _right(nullptr)
, _data(data)
{}
};
template <class T>
class BSTtree
{
typedef BSTNode<T> Node;
public:
BSTtree()
:_root(nullptr)
{}
BSTtree(const BSTtree<T> & r)
{
_root = _Copy(r._root);
}
bool Insert(const T& key)
{
if (_root == nullptr)
{
_root = new Node(key);
return true;
}
Node* parent = nullptr;
Node* pcur = _root;
while (pcur)
{
if (pcur->_data < key)
{
parent = pcur;
pcur = pcur->_right;
}
else if (pcur->_data > key)
{
parent = pcur;
pcur = pcur->_left;
}
else
{
return false;
}
}
pcur = new Node(key);
if (parent->_data > key)
{
parent->_left = pcur;
}
else if (parent->_data < key)
{
parent->_right = pcur;
}
return true;
}
bool Find(const T& key)
{
Node* pcur = _root;
while (pcur)
{
if (key < pcur->_data)
{
pcur = pcur->_left;
}
else if (key > pcur->_data)
{
pcur = pcur->_right;
}
else
{
return true;
}
}
return false;
}
bool Erase(const T& 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
{
// 删除
// 0-1个孩子的情况
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
{
// 2个孩子的情况
// 右子树的最小节点作为替代节点
Node* rightMinP = cur;
Node* rightMin = cur->_right;
while (rightMin->_left)
{
rightMinP = rightMin;
rightMin = rightMin->_left;
}
cur->_key = rightMin->_key;
if (rightMinP->_left == rightMin)
rightMinP->_left = rightMin->_right;
else
rightMinP->_right = rightMin->_right;
delete rightMin;
return true;
}
}
}
return false;
}
bool Inorder()
{
_Inorder(_root);
return true;
}
~BSTtree()
{
_BSTtree(_root);
}
private:
void _Inorder(Node* root)
{
if (root == nullptr)
{
return;
}
_Inorder(root->_left);
printf("%d ", root->_data);
_Inorder(root->_right);
}
void _BSTtree(Node *root)
{
if (root == nullptr)
{
return;
}
_BSTtree(root->_left);
_BSTtree(root->_right);
delete root;
}
Node* _Copy(Node* root)
{
if (root == nullptr)
{
return nullptr;
}
Node* newroot = new Node(root->_data);
newroot->_left = _Copy(root->_left);
newroot->_right = _Copy(root->_right);
return newroot;
}
Node* _root = nullptr;
};
}
KeyValue模型:
namespace keyvalue
{
template <class T,class U>
struct BSTNode
{
pair<T, U> _kv;
BSTNode<T,U>* _left;
BSTNode<T,U>* _right;
BSTNode(const pair<T, U>& kv = pair<T, U>())
:_left(nullptr)
, _right(nullptr)
, _kv(kv)
{}
};
template <class T,class U>
class BSTtree
{
typedef BSTNode<T,U> Node;
public:
BSTtree()
:_root(nullptr)
{}
bool Insert(const pair<T, U>& key)
{
if (_root == nullptr)
{
_root = new Node(key);
return true;
}
Node* parent = nullptr;
Node* pcur = _root;
while (pcur)
{
if (pcur->_kv.first < key.first)
{
parent = pcur;
pcur = pcur->_right;
}
else if (pcur->_kv.first > key.first)
{
parent = pcur;
pcur = pcur->_left;
}
else
{
return false;
}
}
pcur = new Node(key);
if (parent->_kv.first > key.first)
{
parent->_left = pcur;
}
else if (parent->_kv.first < key.first)
{
parent->_right = pcur;
}
return true;
}
bool Find(const pair<T, U>& key)
{
Node* pcur = _root;
while (pcur)
{
if (key.first < pcur->_kv.first)
{
pcur = pcur->_left;
}
else if (key.first > pcur->_kv.first)
{
pcur = pcur->_right;
}
else
{
return true;
}
}
return false;
}
bool Erase(const pair<T, U>& key)
{
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_kv.first < key.first)
{
parent = cur;
cur = cur->_right;
}
else if (cur->_kv.first > key.first)
{
parent = cur;
cur = cur->_left;
}
else
{
// 删除
// 0-1个孩子的情况
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
{
// 2个孩子的情况
// 右子树的最小节点作为替代节点
Node* rightMinP = cur;
Node* rightMin = cur->_right;
while (rightMin->_left)
{
rightMinP = rightMin;
rightMin = rightMin->_left;
}
cur->_kv.first = rightMin->_kv.first;
if (rightMinP->_left == rightMin)
rightMinP->_left = rightMin->_right;
else
rightMinP->_right = rightMin->_right;
delete rightMin;
return true;
}
}
}
return false;
}
bool Inorder()
{
_Inorder(_root);
return true;
}
~BSTtree()
{
_BSTtree(_root);
_root = nullptr;
}
BSTtree(const BSTtree<T, U>& b)
{
_root = _Copy(b._root);
}
private:
void _Inorder(Node* root)
{
if (root == nullptr)
{
return;
}
_Inorder(root->_left);
printf("%d ", root->_kv.first);
_Inorder(root->_right);
}
void _BSTtree(Node* root)
{
if (root == nullptr)
{
return;
}
_BSTtree(root->_left);
_BSTtree(root->_right);
delete root;
}
Node* _Copy(Node* root)
{
if (root == nullptr)
{
return nullptr;
}
Node* newroot = new Node(root->_kv);
newroot->_left = _Copy(root->_left);
newroot->_right = _Copy(root->_right);
return newroot;
}
Node* _root = nullptr;
};
}