1. 二叉搜索树定义
二叉搜索树(Binary Search Tree),又名二叉排序树(Binary Sort Tree)。
二叉搜索树是具有有以下性质的二叉树:
(1)若左子树不为空,则左子树上所有节点的值均小于或等于它的根节点的值。
(2)若右子树不为空,则右子树上所有节点的值均大于或等于它的根节点的值。
(3)左、右子树也分别为二叉搜索树。
2. 二叉搜索树的相关操作
2.1 插入操作
从根节点开始,若插入的值比根节点的值小,则将其插入根节点的左子树;若比根节点的值大,则将其插入根节点的右子树。该操作可使用递归进行实现。
程序代码:
TreeNode* insertBST(TreeNode *root, int k) {
if(root == NULL) {
root = new TreeNode(k);
return root;
}
if(k < root->val) {
root->left = insertBST(root->left, k);
} else if(k > root->val) {
root->right = insertBST(root->right, k);
} else {
return NULL;
}
return root;
}
2.2 查询操作
从根节点开始查找,待查找的值是否与根节点的值相同,若相同则返回True;否则,判断待寻找的值是否比根节点的值小,若是则进入根节点左子树进行查找,否则进入右子树进行查找。该操作使用递归实现。
程序代码:
bool searchBST(TreeNode *root, int k) {
if(root == NULL)
return false;
if(k < root->val)
return searchBST(root->left, k);
else if(k > root->val)
return searchBST(root->right, k);
else
return true;
}
2.3 查找二叉搜索树中的最大(小值)
(1)查找最小值:从根节点开始,沿着左子树一直往下,直到找到最后一个左子树节点,按照定义可知,该节点一定是该二叉搜索树中的最小值节点。
程序代码:
TreeNode* findMinInBST(TreeNode *root) {
if(root != NULL && root->left) {
return findMinInBST(root->left);
} else {
return root;
}
}
(2)查找最大值:从根节点开始,沿着右子树一直往下,直到找到最后一个右子树节点,按照定义可知,该节点一定是该二叉搜索树中的最大值节点。
程序代码:
TreeNode* findMaxInBST(TreeNode *root) {
if(root != NULL && root->right) {
return findMaxInBST(root->right);
} else {
return root;
}
}
2.4 删除节点操作
对二叉搜索树节点的删除操作分为以下三种情况:
(1)待删除节点既无左子树也无右子树:直接删除该节点即可
(2)待删除节点只有左子树或者只有右子树:将其左子树或右子树根节点代替待删除节点
(3)待删除节点既有左子树也有右子树:找到该节点右子树中最小值节点,使用该节点代替待删除节点,然后在右子树中删除最小值节点。
程序:
TreeNode* delTreeNode(TreeNode *root, int k) {
if(root == NULL)
return NULL;
if(k < root->val) {
root->left = delTreeNode(root->left, k);
} else if(k > root->val) {
root->right = delTreeNode(root->right, k);
} else {
if (root->left != NULL && root->right != NULL) {
TreeNode *temp = findMinInBST(root->right);
root->val = temp->val;
root->right = delTreeNode(root->right, temp->val);
} else if(root->left == NULL && root->right == NULL) {
delete root;
root = NULL;
} else if (root->left == NULL) {
TreeNode *temp = root;
root = root->right;
delete temp;
} else if(root->right == NULL) {
TreeNode *temp = root;
root = root->left;
delete temp;
}
}
return root;
}
其他程序:
TreeNode* createBST(vector<int> nums) {
if(nums.empty())
return NULL;
TreeNode *root = new TreeNode(nums[0]);
for(int i = 1; i < nums.size(); i++) {
insertBST(root, nums[i]);
}
return root;
}
void inOrderBST(TreeNode *root) {
if(root == NULL)
return;
inOrderBST(root->left);
cout << root->val << " ";
inOrderBST(root->right);
// cout << endl;
}