详解二叉搜索树(BST)的Java实现和五种遍历方式，移动智能终端开发技术第三次作业

//记作当前节点

TreeNode current = root;

//current为null，跳出循环，parent节点代表我们要插入元素的父节点

while (current != null) {

//如果我们要插入的节点小于当前节点，就去遍历其左子树

//否则遍历右子树

if (o.compareTo(current.element) < 0) {

parent = current;

current = current.left;

} else if (o.compareTo(current.element) > 0) {

parent = current;

current = current.right;

} else {

//树中存在当前要插入的元素，则插入失败

return false;

}

}

//比较当前元素和父节元素的大小

if (o.compareTo(parent.element) < 0) {

//小于父节点的元素，则新建元素节点作为父节点的左节点

parent.left = new TreeNode<>(o);

} else {

//大于父节点的元素，
则新建元素节点作为父节点的右节点

parent.right = new TreeNode<>(o);

}

}

//当前树种存储的元素+1

size++;

return true;

}

4.查找一个元素

---

当我们需要在BST中查找一个节点时，就从根节点从下扫描，直到找到匹配的元素或者达到一个空子树(树中不存在当前要查找的元素)，下面我们来看下它的Java实现：

//返回true表示查询成功，false表示没右查询到

public boolean search(E o) {

//记录遍历时的当前节点

TreeNode current = root;

while (current != null) {

//小于当前节点元素遍历其左子树

if (o.compareTo(current.element) < 0) {

current = current.left;

/大于当前节点元素遍历其右子树

} else if (o.compareTo(current.element) > 0) {

current = current.right;

} else {

//查到

return true;

}

}

return false;

}

5.删除一个元素

---

当我们要在BST中删除一个元素，首先需要定位该元素的位置。然后在删除它时，比插入的时候要复杂一点，因为需要考虑两种情况：

1.当前元素节点没有左子节点。

2.当前元素节点右左子节点。

下面我们直接从代码角度来看看如何来删除一个节点：

//删除成功返回true,否则返回false

public boolean delete(E o) {

//指向当前节点的父节点

TreeNode parent = null;

//指向当前节点

TreeNode current = root;

//遍历二叉树

while (current != null) {

if (o.compareTo(current.element) < 0) {

parent = current;

current = current.left;

} else if (o.compareTo(current.element) > 0) {

parent = current;

current = current.right;

} else {

//找到跳出循环 current就是要删除的节点

break;

}

}

//如果为null返回false

if (current == null) {

return false;

}

//如果要删除的节点没有左节点

if (current.left == null) {

//如果当前要删除节点是根节点，当前节点的右子节点作为根节点。

if (parent == null) {

root = current.right;

} else {

//比较当前要删除的元素和父节点元素的大小

//如果当前要删除的元素是父节点的左子节点

//当前节点的右子节点作为父节点的左子节点

if (o.compareTo(parent.element) < 0) {

parent.left = current.right;

} else {

//如果当前要删除的元素是父节点的右子节点

//当前节点的右子节点作为父节点的右子节点

parent.right = current.right;

}

}

} else {

//如果存在要删除的节点存在左子节点,找到current的左子树的最大元素节点

//parentOfRightMost 指向 最大元素节点的父节点

//最大元素节点不会存在右节点只会存在左节点(如果存在，就不是最大了)

TreeNode parentOfRightMost = current;

TreeNode rightMost = current.left;

while (rightMost.right != null) {

parentOfRightMost = rightMost;

rightMost = rightMost.right;

}

//最大元素节点替换要删除的节点
//最大元素节点不会存在右节点只会存在左节点(如果存在，就不是最大了)

TreeNode parentOfRightMost = current;

TreeNode rightMost = current.left;

while (rightMost.right != null) {

parentOfRightMost = rightMost;

rightMost = rightMost.right;

}

//最大元素节点替换要删除的节点