树的删除
要求:
- 如果删除的是叶子节点,则删除该节点
- 如果删除的是非叶子节点,则删除改字数
- 测试,删除调5号叶子节点和3号子树
任务:
完成删除节点的操作
规定:
如果删除的是叶子节点,则删除该节点
如果删除的是非叶子节点,则删除改字数
思路
- 因为我们的二叉树是单向的,所以我们是判断当前节点的子节点是否需要删除,而不能去判断当前这个节点是不是需要删除节点。
- 如果当前节点的左子树不为空,并且左子节点就是要删除的节点this.left = null;结束递归删除。
- 如果当前节点的右子树不为空,并且右子节点就是要删除的节点this.right = null;结束递归删除
- 如果第二、第三步没有删除节点,那么我们就需要想左子树进行递归删除
- 如果第四步也没有删除这个节点,则当前向右子树进行递归删除
- 考虑如果树是空树的情况,或者只有一个root节点,则等价于将二叉树置空
二叉树
// 删除节点
public void deleteTreeNode(int no) {
if (root != null) {
//必须马上判断是不是要删除的节点
if (root.getNo() == no) root = null;
else root.deleteTreeNode(no);
}else System.out.println("空树,不能删除");
}
递归方法
// 删除二叉树
public void deleteTreeNode(int no) {
// 2. 如果当前节点的左子树不为空,并且左子节点就是要删除的节点this.left = null;结束递归删除。
if (this.left != null && this.left.no == no){
this.left = null;
return;
}
// 3. 如果当前节点的右子树不为空,并且右子节点就是要删除的节点this.right = null;结束递归删除
if (this.right != null && this.right.no == no) {
this.right = null;
return;
}
// 4. 如果第二、第三步没有删除节点,那么我们就需要想左子树进行递归删除
if (this.left != null) this.left.deleteTreeNode(no);
// 5. 如果第四步也没有删除这个节点,则当前向右子树进行递归删除
if (this.right != null) this.right.deleteTreeNode(no);
}