二叉搜索树的相关操作

二叉搜索树的概念

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

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

二叉搜索树的数据结构（同普通二叉树）

typedef struct TreeNode
{
	char data;
	struct TreeNode* lchild;
	struct TreeNode* rchild;
};

二叉搜索树的创建

template<class T>   //中序遍历是递增序列(检测时候使用)
TreeNode* SortBinaryTree(TreeNode* root, const T& val)
{
	if (root == NULL)
	{ 
		root = new TreeNode(val);    //根节点为空，插入，返回。
		return root;
	}
	else  
	{
		if (val <= root->data)  //小于等于根节点，递归插入左子树
		{
			root->lchild = SortBinaryTree(root->lchild, val);
		}
		else   //大于根节点，递归插入右子树
		{
			root->rchild = SortBinaryTree(root->rchild, val);
		}
	}
	return root;
}

二叉搜索树的查找

template<class T>
TreeNode* find(TreeNode* root, const T&key)
{
	TreeNode* cur = root;
	while (cur != NULL&&cur->data != key)  //当查找元素不为空且与根节点不同
	{
		if (key < cur->data)    //小于根节点左子树找
		{
			cur = cur->lchild;
		}
		else   //大于根节点右子树找
		{
			cur = cur->rchild;
		}
	}
	if (cur == NULL)  //若cur=NULL则说明遍历完没有找到，
		return NULL;
	return cur;  
}

二叉搜索树的插入

template<class T>
void insert(TreeNode* root, const T& val)
{
	if (root == NULL)  //根节点为空，直接插入，返回。
	{
		root = new TreeNode(val);
		return;
	}
	TreeNode* cur = root;
	TreeNode* parent = NULL;
	bool isLeftchild = true;  //设置flag判断插入的位置在左子树还是右子树中

	while (cur != NULL) //遍历查找空位置
	{
		parent = cur; 
		if (val < cur->data)  //插入值小于根节点
		{
			cur = cur->lchild;  //在左子树中
			isLeftchild = true;
		}
		else  //插入值大于根节点
		{
			cur = cur->rchild;  //在右子树中
			isLeftchild = false;
		}  
	}
	TreeNode* new_node = new TreeNode(val);//创建新节点
	if (isLeftchild)   //根据标志位判断插入位置是否在左子树
	{
		parent->lchild=new_node;   //插入当前节点的左孩子
	}
	else
	{
		parent->rchild = new_node;  //插入当前节点的右孩子
	}
}

二叉搜索树的删除

template<class T>
void delete(TreeNode* root, const T& val)
{
    if (root == NULL) return ;
    if (data < root->data) delete(root->lchild, data);
    else if (data > root->data) delete(root->rchild, data);
    else {
        TreeNode* cur = root;
        if (root->lchild == NULL) {
            root = root->rchild;
            free(cur);
            cur = NULL;
        }
        else if (root->rchild == NULL) {
            root = root->lchild;
            free(cur);
            cur = NULL;
        }
        else {
            TreeNode* del = root->rchild;
            while (del->lchild) del = del->lchild;
            root->data = del->data;
            delete(root->rchild, del->data);
        }
    }
}