二叉树删除结点

删除要求

由于我现在所学习的二叉树并不是二叉有序树所以对二叉树的删除条件做了些变动：

• 如果删除的结点是叶子结点，则删除该结点
• 如果删除的结点是非叶子结点，则删除该子树

思路分析

1、如果该树只有一个根结点 那么将树置空
2、如果当前结点的左子结点不为空并且左子结点就是要删除的结点就将当前结点的左子结点置空并且返回结束递归
3、如果当前结点的右子结点不为空并且右子结点就是要删除的结点就将当前结点的右子结点置空并且返回结束递归
4、如果当前结点的左右子节点并不是要删除的结点，则向左子树递归删除
5、如果左子树中没有要删除的结点就向右子树递归删除

代码实现

/**
 * 模拟结点
 */
public class Node {
    private int id;
    private String name;
    /*左节点*/
    private Node left;
    /*右节点*/
    private Node right;

    /*构造方法*/
    public Node(int id, String name) {
        this.id = id;
        this.name = name;
    }
    /*前序遍历*/
    public void preDisplay(){
        System.out.println(this);
        if (this.left != null) {
            this.left.preDisplay();
        }
        if (this.right != null) {
            this.right.preDisplay();
        }

    /**
     * 根据id删除结点
     * 说明：
     *  这里删除的如果是叶子结点就删除该节点
     *  如果删除的不是叶子节点就删除那个结点及其子树
     * @param id
     * @return true表示删除成功 false表示没找到结点
     */
    public boolean delete(int id){
        if (this.left != null && this.left.id == id) {
            /*如果当前结点的左子结点不为空并且是要删除的结点就删除左子结点*/
            this.left = null;
            return true;
        }
        if (this.right != null && this.right.id == id) {
            /*如果当前结点的右子结点不为空并且是要删除的结点就删除右子结点*/
            this.right = null;
            return true;
        }
        /*当前结点的左子节点和右子节点都不是要删除是的结点*/
        if (this.left != null) {
            //向左子树递归删除
            this.left.delete(id);
        }
        if (this.right != null){
            //向右子树递归删除
            boolean delete = this.right.delete(id);
            if (delete){
                return delete;
            }
        }
        return false;
    }

	getter and setter 方法

    @Override
    public String toString() {
        return "[" + "id=" + id + ", name=" + name  + ']';
    }
}

public class BinaryTree {

    /*根结点*/
    private Node root;

    public void setRoot(Node root) {
        this.root = root;
    }
    /*前序遍历*/
    public void preShow(){
        if (this.root != null) {
            this.root.preDisplay();
        }else{
            System.out.println("二叉树为空");
        }
    }

    /*删除结点*/
    public void delNode(int id){
        if (root != null) {
            /*判断根节点是否为要删除的结点*/
            if (root.getId() == id) {
                root = null;
                System.out.println("删除成功");
            }else{
                boolean isDelete = root.delete(id);
                System.out.println(isDelete ? "删除成功" : "删除失败没找到结点");
            }
        }else{
            System.out.println("二叉树为空");
        }
    }
}

public class BinaryTreeTest {
    public static void main(String[] args) {
        /*手动创建结点*/
        Node node1 = new Node(1,"张三");
        Node node2 = new Node(2,"李四");
        Node node3 = new Node(3,"王五");
        Node node4 = new Node(4,"赵六");
        node1.setLeft(node2);
        node1.setRight(node3);
        node3.setRight(node4);

        /*创建二叉树*/
        BinaryTree tree = new BinaryTree();
        tree.setRoot(node1);

        System.out.println("删除前的前序遍历：");
        tree.preShow();
        tree.delNode(4);
        System.out.println("删除后的前序遍历：");
        tree.preShow();
    }
}

运行结果

删除前的前序遍历：
[id=1, name=张三]
[id=2, name=李四]
[id=3, name=王五]
[id=4, name=赵六]
删除成功
删除后的前序遍历：
[id=1, name=张三]
[id=2, name=李四]
[id=3, name=王五]

Process finished with exit code 0