二叉排序树的创建和删除
最近学习了二叉树的创建和删除功能,本文代码主要实现以下功能:
- 给定一个数列,创建二叉排序树(BST)
- 遍历二叉排序树(中序遍历)
- 删除二叉排序树的节点
其中,二叉排序树的删除节点步骤较为繁琐,思路总结如下:
一.如果要删除的节点是叶子节点
- 1.找到要删除节点:targetNode;
- 2.找到要删除节点的父节点:parent;
- 3.判断要删除的节点是其父节点的左子节点还是右子节点;
- 4.parent.left=null或者parent.right=null;
二.如果要删除的节点只有一棵子树
- 1.找到要删除节点:targetNode;
- 2.找到要删除节点的父节点:parent;
- 3.如果要删除的节点是其父节点的左子节点:
3.1 要删除节点的子节点是左子节点:parent.left=targetNode.left;
3.2 要删除节点的子节点是右子节点:parent.left=targetNode.right; - 4.如果要删除的节点是其父节点的右子节点:
4.1 要删除节点的子节点是左子节点:parent.right=targetNode.left;
4.2 要删除节点的子节点是右子节点:parent.right=targetNode.right;
三.如果要删除的节点有两棵子树
- 第一种方案:找到要删除节点左子树中最大的值;
- 第二种方案:找到要删除节点右子树中最小的值;
以第二种方案为例:
- 1.找到要删除节点:targetNode;
- 2.找到要删除节点的父节点:parent;
- 3.找到要删除节点右子树中最小值对应的节点,将其保存:temp
- 4.删除该最小节点
- 5.将要删除的节点替换成该最小节点targetNode.value=temp.value
如有错误之处,还望指出,定会及时改正
package com.binarySortTree;
/*
* 1.给定一个数列,创建二叉排序树(BST)
* 2.遍历二叉排序树(中序遍历)
* 3.删除二叉排序树的节点
*/
public class BinarySortTreeDemo2 {
public static void main(String[] args) {
int[] arr = { 8, 5, 10, 6, 3, 2, 4, 7, 9, 1 };
BinarySortTree2 binarySortTree2 = new BinarySortTree2();
// 创建二叉排序树
for (int i = 0; i < arr.length; i++) {
binarySortTree2.add(new Node2(arr[i]));
}
// 中序遍历
System.out.println("中序遍历二叉排序树");
binarySortTree2.infixList();
//测试删除节点
binarySortTree2.delNode2(1);
/*binarySortTree2.delNode2(2);
binarySortTree2.delNode2(3);
binarySortTree2.delNode2(9);
binarySortTree2.delNode2(7);
binarySortTree2.delNode2(8);
binarySortTree2.delNode2(5);
binarySortTree2.delNode2(10);
binarySortTree2.delNode2(4);
binarySortTree2.delNode2(6);*/
System.out.println("删除节点后");
binarySortTree2.infixList();
}
}
//创建二叉排序树类
class BinarySortTree2 {
Node2 root;
/**
* 创建二叉排序树(添加节点)
*
* @param node
*/
public void add(Node2 node) {
if (root == null) {
root = node;
} else {
root.add(node);
}
}
/**
* 中序遍历二叉树
*/
public void infixList() {
if (root == null) {
System.out.println("空树无法遍历!");
} else {
root.infixList();
}
}
// 删除节点:三种情况
/*
* 一.如果要删除的节点是叶子节点
* 1.找到要删除节点:targetNode;
* 2.找到要删除节点的父节点:parent;
* 3.判断要删除的节点是其父节点的左子节点还是右子节点;
* 4.parent.left=null或者parent.right=null;
* 二.如果要删除的节点只有一棵子树
* 1.找到要删除节点:targetNode;
* 2.找到要删除节点的父节点:parent;
* 3.如果要删除的节点是其父节点的左子节点:
* 3.1 要删除节点的子节点是左子节点:parent.left=targetNode.left;
* 3.2 要删除节点的子节点是右子节点:parent.left=targetNode.right;
* 4.如果要删除的节点是其父节点的右子节点:
* 4.1 要删除节点的子节点是左子节点:parent.right=targetNode.left;
* 4.2 要删除节点的子节点是右子节点:parent.right=targetNode.right;
* 三.如果要删除的节点有两棵子树
* 第一种方案:找到要删除节点左子树中最大的值;
* 第二种方案:找到要删除节点右子树中最小的值;
* 以第二种方案为例:
* 1.找到要删除节点:targetNode;
* 2.找到要删除节点的父节点:parent;
* 3.找到要删除节点右子树中最小值对应的节点,将其保存:temp
* 4.删除该最小节点
* 5.将要删除的节点替换成该最小节点targetNode.value=temp.value
*/
/**
* 删除节点
*
* @param value
*/
public void delNode2(int value) {
if (root == null) {
System.out.println("二叉排序树为空!");
return;
}
// 首先要找到要删除的节点
Node2 targetNode = root.search(value);
if (targetNode == null) {
System.out.println("没有找到要删除的节点!");
return;
}
// 如果要删除的节点就是根节点,并且该二叉树只有一个节点,直接将根节点置为空即可!
if (targetNode.value == root.value && root.left == null && root.right == null) {
root = null;
return;
}
// 找到要删除节点的父节点
Node2 parent = root.searchParent(value);
// 一.如果要删除的节点是叶子节点
if (targetNode.left == null && targetNode.right == null) {
// 1.如果要删除节点是其父节点的左子节点
if (parent.left == targetNode) {
parent.left = null;
} else {// 2.如果要删除节点是其父节点的右子节点
parent.right = null;
}
return;
}
// 二.如果要删除的节点只有一棵子树--左子树
if (targetNode.left != null && targetNode.right == null) {
// !!!这里需要考虑一种特殊情况:如果要删除的节点是根节点,并且根节点下还有个左子节点
if (parent != null) {
// 1.如果要删除的节点是其父节点的左子节点
if (parent.left == targetNode) {
parent.left = targetNode.left;
}
// 2.如果要删除的节点是其父节点的右子节点
if (parent.right == targetNode) {
parent.right = targetNode.left;
}
} else {// 如果要删除的节点是根节点,并且根节点下还有个左子节点
root = targetNode.left;
}
return;
}
// 二.如果要删除的节点只有一棵子树--右子树
if (targetNode.left == null && targetNode.right != null) {
// !!!这里需要考虑一种特殊情况:如果要删除的节点是根节点,并且根节点下还有个右子节点
if (parent != null) {
// 1.如果要删除的节点是其父节点的左子节点
if (parent.left == targetNode) {
parent.left = targetNode.right;
}
// 2.如果要删除的节点是其父节点的右子节点
if (parent.right == targetNode) {
parent.right = targetNode.right;
}
} else {// 如果要删除的节点是根节点,并且根节点下还有个右子节点
root = targetNode.right;
}
return;
}
// 三.如果要删除的节点有两棵子树(第二种方案:找要删除节点的右子树中最小值的节点)
if (targetNode.left != null && targetNode.right != null) {
// 1.在要删除节点的右子树中找最小值的节点
int min = searchNodeTreeMin(targetNode.right);
// 2.将要删除节点替换成最小值的节点
targetNode.value = min;
return;
}
}
/**
* 根据传入的node值,将node认定为根节点,找其子树中最小值并将其删除
*
* @param node
* @return
*/
public int searchNodeTreeMin(Node2 node) {
Node2 temp = node;
// 找最左边的叶子节点
while (temp.left != null) {
temp = temp.left;
}
// 将最小值的节点删除
delNode2(temp.value);
// 将最小值返回
return temp.value;
}
}
//创建节点类
class Node2 {
int value;
Node2 left;
Node2 right;
public Node2(int value) {
this.value = value;
}
@Override
public String toString() {
return "Node2 [value=" + value + "]";
}
/**
* 添加节点
*
* @param node
*/
public void add(Node2 node) {
if (node == null) {
return;
}
// 1.要添加的节点的值小于当前节点的值
if (node.value < this.value) {
// 如果当前节点的的左子节点为空,就直接挂上
if (this.left == null) {
this.left = node;
} else {// 如果不为空,左递归
this.left.add(node);
}
} else {// 2.要添加的节点的值大于或等于当前节点的值
// 如果当前节点的右子节点为空,就直接挂上
if (this.right == null) {
this.right = node;
} else {// 如果不为空,右递归
this.right.add(node);
}
}
}
/**
* 中序遍历
*/
public void infixList() {
if (this.left != null) {
this.left.infixList();
}
System.out.println(this);
if (this.right != null) {
this.right.infixList();
}
}
/*
* 查找要删除的节点
*/
public Node2 search(int value) {
if (this.value == value) {// 当前节点就是要删除的节点
return this;
} else if (value < this.value) {// 如果要删除节点的值小于当前节点的值
// 左递归查找
if (this.left == null) {
return null;
} else {
return this.left.search(value);
}
} else {// 如果要删除节点的值大于或等于当前节点的值
if (this.right == null) {
return null;
} else {
return this.right.search(value);
}
}
}
/*
* 查找要删除节点的父节点
*/
public Node2 searchParent(int value) {
// 如果当前节点的子节点中有要删除的节点,那么当前节点就是要删除节点的父节点
if ((this.left != null && this.left.value == value) || (this.right != null && this.right.value == value)) {
return this;
}else {
if (value < this.value && this.left != null) {// 如果要删除节点的值小于当前节点的值,左递归查找
return this.left.searchParent(value);
} else if (value >= this.value && this.right != null) {// 如果要删除节点的值大于或等于当前节点的值,右递归查找
return this.right.searchParent(value);
} else {// 如果都没找到,那就说明要删除的节点没有父节点,返回null
return null;
}
}
}
}