二叉树删除节点

二叉树—删除节点

1）删除节点是叶子节点，删除该节点

2）删除节点是非叶子节点，则删除该子树

思路：　　

　　1、考虑若树是空树root，如果只有一个root节点，则等价将二叉树置空

　　2、因二叉树是单向的，所以判断当前节点的子节点是否需要删除，而不能判断当前这个节点是不是需要删除节点

　　3、如果当前节点的左子节点不为空，并且左子节点就是要删除节点，就将this.left =null；

并且就返回（结束递归删除）

　　4、如果当前节点的右子节点不为空，并且右子节点就是要删除节点，就将this.right =null；

 并且就返回（结束递归删除）

　　5、若第2和第3步没有删除节点，需要向左子树进行递归删除

　　6、若第4步也没有删除节点，向右子树进行递归删除

　　

public class BinaryTreeDemo {

    public static int PRECOUNT = 0;
    public static int INFIXCOUNT = 0;
    public static int POSTCOUNT = 0;
    public static void main(String[] args) {
        //先需要创建一棵二叉树
        BinaryTree binaryTree = new BinaryTree();
        EmpNode root = new EmpNode(1, "aa");
        EmpNode emp2 = new EmpNode(2, "bb");
        EmpNode emp3 = new EmpNode(3, "cc");
        EmpNode emp4 = new EmpNode(4, "dd");
        EmpNode emp5 = new EmpNode(5, "ee");

        //手动创建二叉树，后面学习递归的方式创建二叉树
        root.setLeft(emp2);
        root.setRight(emp3);
        emp3.setRight(emp4);
        emp3.setLeft(emp5);

        //测试
        System.out.println("前序遍历");
        binaryTree.setRoot(root);
        binaryTree.preOrder();

        System.out.println("中序遍历");
        binaryTree.setRoot(root);
        binaryTree.infixOrder();


        System.out.println("后序遍历");
        binaryTree.setRoot(root);
        binaryTree.postOrder();

        //前序查找
        //前序查找次数
        System.out.println("前序查找方式");
        EmpNode resNode = binaryTree.preOrderSearch(5);
        if (resNode != null){
            System.out.printf("找到，信息为no=%d name=%s 查询次数为%s",resNode.getNo(),resNode.getName(),PRECOUNT);

        }else {
            System.out.printf("没找到 no=%d",5);
        }

        System.out.println();

        //中序查找
        //中序查找次数
        System.out.println("中序查找方式");
        EmpNode resNode1 = binaryTree.infixOrderSearch(5);
        if (resNode1 != null){
            System.out.printf("找到，信息为no=%d name=%s 查询次数为%s",resNode1.getNo(),resNode1.getName(),INFIXCOUNT);

        }else {
            System.out.printf("没找到 no=%d",5);
        }

        System.out.println();

        //后序查找
        //后序查找次数
        System.out.println("后序查找方式");
        EmpNode resNode2 = binaryTree.postOrderSearch(5);
        if (resNode1 != null){
            System.out.printf("找到，信息为no=%d name=%s 查询次数为%s",resNode2.getNo(),resNode2.getName(),POSTCOUNT);

        }else {
            System.out.printf("没找到 no=%d",5);
        }




    }

}

//2.定义一个BinaryTree
class BinaryTree {
    private EmpNode root;

    public void setRoot(EmpNode root) {
        this.root = root;
    }

    //前序遍历
    public void preOrder() {
        if (this.root != null) {
            this.root.preOrder();
        } else {
            System.out.println("二叉树为空，无法遍历");
        }
    }

    //中序遍历
    public void infixOrder() {
        if (this.root != null) {
            this.root.infixOrder();
        } else {
            System.out.println("二叉树为空，无法遍历");
        }
    }

    //后序遍历
    public void postOrder() {
        if (this.root != null) {
            this.root.postOrder();
        } else {
            System.out.println("二叉树为空，无法遍历");
        }
    }

    //前序查找
    public EmpNode preOrderSearch(int no){
        if (root != null){
            return root.preOrderSearch(no);
        }else {
            return null;
        }
    }

    //中序查找
    public EmpNode infixOrderSearch(int no){
        if (root != null){
            return root.infixOrderSearch(no);
        }else {
            return null;
        }
    }
    //后序查找
    public EmpNode postOrderSearch(int no){
        if (root != null){
            return root.postOrderSearch(no);
        }else {
            return null;
        }
    }

    //删除节点
    public void delNode(int no){
        if (root != null){
            //如果只有一个root节点，需要立即判断是不是要删除节点
            if (root.getNo() == no){
                root = null;
            }else{
                root.delNode(no);
            }

        }else {
            System.out.println("空树，不能删除");
        }
    }
}

//1.先创建HeroNode
class EmpNode {
    private int no;
    private String name;
    private EmpNode left;//默认为null
    private EmpNode right;//默认为null

    public EmpNode(int no, String name) {
        this.no = no;
        this.name = name;
    }

    public int getNo() {
        return no;
    }

    public void setNo(int no) {
        this.no = no;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public EmpNode getLeft() {
        return left;
    }

    public void setLeft(EmpNode left) {
        this.left = left;
    }

    public EmpNode getRight() {
        return right;
    }

    public void setRight(EmpNode right) {
        this.right = right;
    }

    @Override
    public String toString() {
        return "EmpNode{" +
                "no=" + no +
                ", name='" + name + '\'' +
                '}';
    }

    //递归删除节点
    //1.若删除的节点是叶子节点，则删除该节点
    //2.若删除的节点是非叶子节点，则删除该子树
    public void delNode(int no){
        //思路
        if (this.left != null && this.left.no == no){
            this.left = null;
            return;
        }
        if (this.right != null && this.right.no == no){
            this.right = null;
            return;
        }

        if (this.left != null){
            this.left.delNode(no);
            //return;写了return有可能下面向右不执行了
        }
        if (this.right != null){
            this.right.delNode(no);
        }
    }

    //编写前序遍历方法

    public void preOrder() {
        System.out.println(this);//先输出父节点
        //递归向左子树前序遍历
        if (this.left != null) {
            this.left.preOrder();
        }
        //递归向右子树前序遍历
        if (this.right != null) {
            this.right.preOrder();
        }

    }

    //编写中序遍历方法
    public void infixOrder() {
        //递归向左子树中序遍历
        if (this.left != null) {
            this.left.infixOrder();//2,1,3,4
        }
        //输出父节点
        System.out.println(this);
        //递归向右子树中序遍历
        if (this.right != null) {
            this.right.infixOrder();
        }
    }

    //编写后序遍历方法
    public void postOrder() {
        if (this.left != null) {
            this.left.postOrder();
        }
        if (this.right != null) {
            this.right.postOrder();
        }
        System.out.println(this);
    }

    /**
     * @param no 查找no
     * @return 若找到就返回该Node，若没找到返回Null
     */
    //前序遍历查找
    public EmpNode preOrderSearch(int no) {
        BinaryTreeDemo.PRECOUNT++;
        //System.out.println("进入前序查找");

        //比较当前节点是不是
        if (this.no == no) {
            return this;
        }
        //判断当前节点的左子节点是否为空，若不为空，则递归前序查找
        //若左递归前序查找，找到节点，则返回

        EmpNode resNode = null;
        if (this.left != null) {
            resNode = this.left.preOrderSearch(no);
        }
        if (resNode != null) {
            return resNode;
        }

        //左递归前序查找，找到节点，则返回，否继续判断
        //当前的结点的右子节点是否为空，若为空，则继续向右递归前序查找
        if (this.right != null) {
            resNode = this.right.preOrderSearch(no);
        }
        return resNode;

    }

    //中序遍历查找
    public EmpNode infixOrderSearch(int no) {

        //判断当前节点的左子节点是否为空，若不为空，则递归中序查找
        EmpNode resNode = null;
        if (this.left != null) {
            resNode = this.left.infixOrderSearch(no);
        }
        if (resNode != null) {
            return resNode;
        }
        BinaryTreeDemo.INFIXCOUNT++;
        //若找到，则返回，若没找到，就和当前节点比较，若是则返回
        if (this.no == no) {
            return this;
        }
        //否则继续进行右递归的中序查找
        if (this.right != null) {
            resNode = this.right.infixOrderSearch(no);
        }
        return resNode;

    }

    //后序遍历查找
    public EmpNode postOrderSearch(int no) {


        //判断当前节点的左子节点是否为空，若不为空，则继续递归后序查找
        EmpNode resNode = null;
        if (this.left != null) {
            resNode = this.left.postOrderSearch(no);
        }
        if (resNode != null) {//说明左子树找到
            return resNode;
        }
        //若左子树没找到，则向右子树递归进行后序遍历查找
        if (this.right != null) {
            resNode = this.right.postOrderSearch(no);
        }
        if (resNode != null) {
            return resNode;
        }
        //若左右子树都没找到，就比较当前节点是不是
        BinaryTreeDemo.POSTCOUNT++;
        if (this.no == no) {
            return this;
        }
        return resNode;
    }

}