java有序二叉树的删除节点方式

 更新时间:2024年12月17日 08:57:34   作者:Sshm_666  
文章描述了在二叉树中删除节点的三种情况及其对应的操作步骤,通过递归找到节点及其父节点,并根据节点的子树情况(无子树、单子树、双子树)进行相应的删除操作,文章还提供了一个测试类来验证删除操作的正确性

java有序二叉树的删除节点

删除节点时会有三种可能

1、删除的节点为叶子节点

我们如果删除为叶子节点,则步骤应该是:

1)先找到要删除的叶子节点

2)再找到要删除节点的父节点(考虑是否有父节点)

3)找到删除的节点是父节点的左子树还是右子树

4)根据前面的情况进行删除

2、删除的节点只有一个子树

步骤:

1)先找到要删除的节点

2)再找到要删除节点的父节点(考虑是否有父节点)

3)确定删除的节点是有左子树还是有右子树

4)找到删除的节点是父节点的左子树还是右子树

5)根据前面的情况进行删除

3、删除的节点有两个子树

步骤:

1)先找到要删除的节点

2)再找到要删除节点的父节点(考虑是否有父节点)

3)找到删除节点右子树当中最小的或左子树最大的节点

4)将3)找到的这个节点的值替换掉要删除的节点的值

5)删除找到的3)

通过上面步骤它们都有一个共同特点就是需要找到删除的节点和要删除节点的父节点,所以我们可以把这两个找节点都写成方法:

找到删除节点代码如下(通过递归的方式找到要删除节点位置并记录下来):

//找到要删除的节点
	public TreeNode searchDelnode(TreeNode node,Integer val) {
		if(node==null) {
			return null;
		}
		if(node.val==val) {
			return node;
		}else if(val>node.val){
			if(node.rightNode==null) {
				return null;
			}
			return searchDelnode(node.rightNode, val);
		}else {
			if(node.leftNode==null) {
				return null;
			}
			return searchDelnode(node.leftNode, val);
		}
	}

找被删除节点的父节点代码如下所示:

//找到要删除节点的父节点
	public TreeNode searchDelpartent(TreeNode node,Integer val) {
		if(node==null) {
			return null;
		}
		if((node.leftNode!=null&&node.leftNode.val==val)||(node.rightNode!=null&&node.rightNode.val==val)) {
			return node;
		}else {
			if(node.leftNode!=null&&val<node.val) {
				return searchDelpartent(node.leftNode, val);
			}else if(node.rightNode!=null&&val>node.val){
				return searchDelpartent(node.rightNode, val);
			}else {
				return null;
			}
		}
	}

我们可以根据以上步骤写删除方法代码:

//二叉树删除节点
	public void del(TreeNode node,Integer val) {
		if(node==null) {
			return;
		}
		//1.找到要删除的节点
		TreeNode targetNode=searchDelnode(node, val);
		//2.没有找到要删除的节点
		if(targetNode==null) {
			return;
		}
		//3.找到要删除节点的父节点
		TreeNode parentNode=searchDelpartent(node, val);
		if(node.rightNode==null&&node.leftNode==null) {
			root=null;
			return;
		}
		if(parentNode==null&&(targetNode.rightNode!=null||targetNode.leftNode!=null)) {
			int min=searchRightMin(targetNode.rightNode);
			targetNode.val=min;
			return;
		}
		if(targetNode.leftNode==null&&targetNode.rightNode==null) {
			if(parentNode.rightNode!=null&&parentNode.rightNode.val==targetNode.val) {
				parentNode.rightNode=null;
			}else if(parentNode.leftNode!=null&&parentNode.leftNode.val==val){
				parentNode.leftNode=null;
			}
		}else if(targetNode.leftNode!=null&&targetNode.rightNode!=null) {
//在删除节点由左右两个子树时,我们选择找删除节点右子树的最小值,我们可以写一个方法,在这个方法里不仅找到最小值,并把这个位置的元素进行删除
			int min=searchRightMin(targetNode.rightNode);
			targetNode.val=min;
		}else {
			if(targetNode.rightNode!=null) {
				if(parentNode.rightNode.val==targetNode.val) {
					parentNode.rightNode=targetNode.rightNode;
				}else {
					parentNode.leftNode=targetNode.rightNode;
				}
			}else{
				if(targetNode.rightNode.val==targetNode.val) {
					parentNode.rightNode=targetNode.leftNode;
				}else {
					parentNode.leftNode=targetNode.leftNode;
				}
			}
		}
	}

在删除节点由左右两个子树时,我们选择找删除节点右子树的最小值,我们可以写一个方法,在这个方法里不仅找到最小值,并把这个位置的元素进行删除

代码如下:

public int searchRightMin(TreeNode node) {
		TreeNode temp=node;
		while(temp.leftNode!=null) {
			temp=temp.leftNode;
		}
		del(root, temp.val);
		return temp.val;
	}
}

写一个测试类进行测试:

public class Test {
	public static void main(String[] args) {
		BinaryTree binaryTree=new BinaryTree();
		binaryTree.insert(1);
		binaryTree.insert(2);
		binaryTree.insert(3);
		binaryTree.insert(4);
		binaryTree.insert(5);
		binaryTree.insertDigui(6, binaryTree.root);
		binaryTree.insertDigui(7, binaryTree.root);
		binaryTree.insertDigui(8, binaryTree.root);
		binaryTree.insertDigui(9, binaryTree.root);
		binaryTree.del(binaryTree.root, 2);
		binaryTree.Order();
		System.out.println();
		binaryTree.startErgodic(binaryTree.root);
		System.out.println();
		binaryTree.midErgodic(binaryTree.root);
		System.out.println();
		binaryTree.endErgodic(binaryTree.root);
	}
}

结果如下图所示:

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

最新评论