尚硅谷-数据结构与算法-二叉树笔记3

主要是删除节点的操作

 

package com.atguigu.tree;

/**
 * 基础知识：节点（一个对象）、根节点（顶）、父节点、子节点、叶子节点（没有子节点得节点）、节点得权（节点得值）、路径（从root节点找到
 * 该节点得路线），层、子树、树得高度（最大层数）、森林（多颗子树构成得森林）
 * 满二叉树：所有得叶子节点都在最后一层，并且系节点得总数=2^n-1，n为层数，则我们称为满二叉树
 * 完全二叉树：如果该二叉树得所有叶子节点都在最后一层或者倒数第二层，而且最后一层得叶子节点在左边连续，倒数第二层得叶子节点在右边连续
 * ，我们称为完全二叉树
 * 前序遍历：先输出父节点，再遍历左子树和右子树
 * 中序遍历：先遍历左子树，再输出父节点，再遍历右子树
 * 后续遍历：先遍历左子树，再遍历右子树，最后输出父节点
 * 小结：看输出父节点得顺序，就确定是前序、中序还是后序
 * 分析：二叉树得前序，中序和后序得步骤
 * 1.首先创建一个二叉树
 * 2.前序遍历
 *   2.1先输出当前节点（初始的时候是根节点，root节点）
 *   2.2如果左子节点不为空，则递归继续前序遍历
 *   2.3如果右子节点不为空，则递归及继续前序遍历
 * 3.中序遍历
 *   3.1如果当前节点得左子节点不为空，则递归中序遍历
 *   3.2输出当前节点
 *   3.3如果当前节点得右子节点不为空，则递归中序遍历
 *4.后续遍历
 *   4.1如果当前节点的左子节点不为空，则递归后序遍历
 *   4.2如果当前节点的右子节点不为空，则递归后续遍历
 *   4.3输出当前节点
 *使用前序、中序、后序的方式来查询指定的节点
 *  前序查找思路
 *  1、先判断当前id是否等于要查找的id
 *  2、如果是相等，则返回当前节点
 *  3、如果不等，则判断当前节点的左子节点是否为空，如果不为空，则递归前序查找
 *  4、如果左递归前序查找，找到了，则返回；否则继续判断，当前节点的右子节点
 *  是否为空，如果不为空，则继续向右递归前序查找
 *
 *  中序查找思路
 *   1、先判断当前节点的左子节点是否为空，如果不为空，则递归中序查找
 *   2、如果如果找到，就返回，如果没有找到，就和当前节点进行比较，如果是（找到）
 *   则返回当前节点，否则继续进行右递归的中序查找。
 *   3、如果右递归中序查找，找到了就返回，找不到就返回null
 *
 *   后序查找思路
 *   1、先判断当前的左子节点是否为空，如果不为空，则递归后序查找
 *   2、如果找到就返回，如果没有找到就判断当前节点的右子节点是否为空，
 *   如果不为空，则右递归进行后序查找，如果找到，就返回
 *   3、如果没有找到，就和当前节点进行比较，如果是（找到）就返回，否则返回null
 *
 *   二叉树的删除节点思路
 *   规定
 *   1.如果删除的是叶子节点，则删除该节点
 *   2.如果删除的节点是非叶子节点，则删除该子树
 *   3.测试，删除5号叶子节点和3号子树
 *思路：
 *   首先：
 *   考虑如果树是空树root,如果只有一个root节点，则等价将二叉树置空。
 *   //然后进行下面的操作
 *    1.因为我们的二叉树是单向的，所以我们是判断当前节点的子节点！！！是否需要删除的节点，
 *    而不能去判断，而不能判断当前这个节点是不是需要删除的节点。（也就是说，如果指针
 *    指向5号，就删除不了5号，只能删除其左右子节点，找不到5号的父节点）
 *    2.如果当前节点的左子节点不为空，并且左子节点的编号就是要删除的节点，那么就直接，
 *    将this.left=null;直接置空即可。并且就返回（结束递归删除）
 *    3.如果当前节点的右子节点不为空，并且右子节点就是要删除的节点，那么直接将
 *    右边置空即可，this.right=null;并且就返回（结束递归删除）
 *    4.如果第2步和第3步，都没有删除节点，那么我们就需要向左子树进行递归删除。（当然
 *    也得判断左子树是否为null空）
 *    5.如果第4步也没有删除节点，则应当向右子树进行递归删除。
 * 先判断1宋江，如果不是判断2吴用（实际上并不是先判断2，而是2的下面，因为2在判断1宋江的时候就
 * 已经判断过了，然后一直判断2的左边知道走完），然后回到1，判断1右边3卢俊义，如果3不是，就判断
 * 3的左边关胜是不是（如果关胜不是继续往下走，直到关胜左边走完，这里关胜下面没有，然后又回到卢俊义）
 * ，然后又判断卢俊义右边是不是，最后整个走完了，有就删除=====================这个分析有错误
 *
 * ==========删除5号正确步骤，先判断1宋江，然后再判断2吴用，然后再判断3卢俊义，3卢俊义也不是，
 * 然后就回到this.left，这个其实是指的2吴用，以吴用节点递归，吴用的左右都是空，此时也尝试向
 * 吴用的左边递归，但是吴用的左边是null所以就进不去，右边也是空，也不进去。然后就回来了回来1号
 * 宋江这个点，然后向宋江的右边进行递归，判断3号卢俊义，进入右递归，进入递归后，先判断左边5号关胜
 * 这个符合、置空return
 *
 * @author WZ
 * @create 2021-10-15 21:37
 */
public class BinaryTreeDemo {
    public static void main(String[] args) {
        //先需要一个二叉树
        BinaryTree binaryTree = new BinaryTree();
        //创建需要的节点
        HeroNode root= new HeroNode(1, "宋江");
        HeroNode node2 = new HeroNode(2, "吴用");
        HeroNode node3 = new HeroNode(3, "卢俊义");
        HeroNode node4 = new HeroNode(4, "林冲");
        HeroNode node5 = new HeroNode(5, "关胜");//再加一个


        //说明：先手动创建该二叉树，后面我们学习递归的方式创建二叉树
        root.setLeft(node2);//root左边的节点挂上了node2，之前是root.left=node2;但是这个方法变成私有的了
        root.setRight(node3);
        node3.setRight(node4);
        node3.setLeft(node5);//
        binaryTree.setRoot(root);

        //测试
        System.out.println("前序遍历");//先输出1宋江，2吴用，3卢俊义，4林冲，正好是顺序
        binaryTree.preOrder();//再加一个5，是12354

        //测试
        System.out.println("中序遍历");
        binaryTree.infixOrder();//2,1,3,4；//再加一个5，是21534

        //后序遍历
        System.out.println("后序遍历");//此时需要注意
        binaryTree.postOrder();//2，4，3，1//再加一个5，是254311

        //前序查找遍历
        //前序遍历的次数4
       /* System.out.println("前序遍历查找的方式~~");
        HeroNode resNode = binaryTree.preOrderSearch(5);
//        System.out.println(resNode);直接输出即可，毕竟都重写了toString，但是老师是按照下面写的
        if (resNode!=null){
            System.out.printf("找到了，信息为id=%d name=%s\n",resNode.getNo(),resNode.getName());
        }else {
            System.out.printf("没有找到id=%d的英雄",5);
        }*/

        /*//中序查找遍历
        //中序遍历的次数3
        System.out.println("中序遍历查找的方式~~");
        HeroNode resNode1= binaryTree.infixOrderSearch(5);
        if (resNode1!=null){
            System.out.printf("找到了，信息为id=%d name=%s",resNode1.getNo(),resNode1.getName());
        }else {
            System.out.printf("没有找到id=%d的英雄",5);
        }*/

        //后序查找遍历
        //后序遍历的次数3
        /*System.out.println("后序遍历查找的方式~~");
        HeroNode resNode2= binaryTree.postOrderSearch(5);
        if (resNode2!=null){
            System.out.printf("找到了，信息为id=%d name=%s",resNode2.getNo(),resNode2.getName());
        }else {
            System.out.printf("没有找到id=%d的英雄",5);
        }*/
        //测试，删除节点的操作
        System.out.println("删除前，前序遍历的操作");
        binaryTree.preOrder();
        binaryTree.delNode(5);
//        binaryTree.delNode(3);//删除子树，等价与把那棵子树全部干掉
        System.out.println("删除后，前序遍历的操作");
        binaryTree.preOrder();//1234
    }
}

//定义BinaryTree 二叉树
class BinaryTree{
    //只需要一个根节点即可
    private HeroNode root;

    public void setRoot(HeroNode root) {
        this.root = root;
    }
    //前序遍历；真正的调用是从根节点来调的，就想前面所讲的哈希表
    public void preOrder(){
        if (this.root!=null){
            this.root.preOrder();//谁调用指向谁preOrder()指向root
        }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 HeroNode preOrderSearch(int no){
        if (root!=null){
          return   root.preOrderSearch(no);
        }else {
            return null;//如果为空的话直接返回即可
        }
    }
    //中序查找遍历
    public HeroNode infixOrderSearch(int no){
        if (root!=null){
           return root.infixOrderSearch(no);
        }else {
            return null;
        }
    }
    //后序遍历查找
    public HeroNode postOrderSearch(int no){
        if (root!=null){
            return root.postOrderSearch(no);
        }else {
            return null;
        }
    }

    //删除节点的操作
    public void delNode(int no){
        //首先判断root是否为空
        if (root!=null){
            //如果只有一个root节点，这里需要立即判断root是不是要删除的节点
            if (root.getNo()==no){
                root=null;//恰好是根节点，直接置空即可
            }else{
                //如果root不是，则就递归删除
                root.delNode(no);
            }
        }else {
            System.out.println("空树，不能删除~~");
        }
    }

}


//先创建HeroNode 节点
class HeroNode{
    private int no;//英雄的编号
    private String name;//英雄的名字
    private HeroNode left;//指向左边的索引，默认null
    private HeroNode right;//指向右边的索引，默认null

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

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

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

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

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

    public int getNo() {
        return no;
    }

    public String getName() {
        return name;
    }

    public HeroNode getLeft() {
        return left;
    }

    public HeroNode getRight() {
        return right;
    }

    //需要输出当前节点的内容，所以需要重写toString方法


    @Override
    public String toString() {
        return "HeroNode{" +
                "no=" + no +
                ", name='" + name + '\'' +
                '}';
    }
    //编写，前序遍历的方法
    public void preOrder(){
        //先输出当前节点，不用判断是否为空，因为都进来了肯定不为空
        System.out.println(this);//先输出父节点，不能写root，后面无法递归，第一次this=root
        //递归向左子树前序遍历
        if (this.left!=null){
            this.left.preOrder();
        }
        //向左递归结束后，返回到root，递归向右子树进行遍历
        if (this.right!=null){
            this.right.preOrder();
        }

    }
    //中序遍历的方法
    public void infixOrder(){
        //先递归，向左子树中序遍历
        if (this.left!=null){
            this.left.infixOrder();
        }
        //输出当前节点：父节点
        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 英雄的编号（id)
     * @return 找到返回该英雄node，没找到返回null
     */
    public HeroNode preOrderSearch(int no){
        System.out.println("进入前序遍历查找的方式");
       //比较当前节点是不是
       if (this.no==no){
           return this;//
       }
       //如果不是，判断左子节点是否为空，不为空就左递归;
        // 如果左递归找到，节点，则返回
        HeroNode resNode=null;//假定我们找到的节点就是它，因为我们需要判断左递归找到要返回
        if (this.left!=null){
           resNode=this.left.preOrderSearch(no);
        }
        if (resNode!=null){//这个说明左子树我们已经找到了
            return resNode;
        }
        //左递归没有找到的话，需要继续判断
        //判断当前节点的右子节点是否为空，如果不为空，则继续向右递归查找，找到返回，没有返回null
        if (this.right!=null){
            resNode=this.right.preOrderSearch(no);
        }
        return resNode;//不管找没找到，就可以直接返回

    }
    //中序遍历查找
    public HeroNode infixOrderSearch(int no){
        //先判断当前节点的左子节点是否为空，如果不为空，则递归中序查找
        HeroNode resNode=null;
        if (this.left!=null){
            resNode=this.left.infixOrderSearch(no);
        }
        if (resNode!=null){//这个表示左节点找到
            return resNode;
        }

        System.out.println("进入zhong序遍历查找");

        //如果没有找到，就和当前节点比较，如果是，则返回当前节点
        if (this.no==no){
            return this;//说明找到，并返回了当前节点
        }
        //如果没有找到，就向右继续递归中序查找
        if (this.right!=null){
            resNode=this.right.infixOrderSearch(no);
        }
        return resNode;//不管是否为空，直接返回即可
    }
    //后序遍历查找
    public HeroNode postOrderSearch(int no){
        //判断当前节点的左子节点是否为空，如果不为空，则递归后序查找
        HeroNode 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;
        }

        System.out.println("进入后序查找");//查看遍历的次数，得写在比较语句(this.no==no)的前面，不能写在前面

        //如果左右子树都没有找到，就比较当前节点是不是
        if (this.no==no){
            return this;

        }
        return resNode;
    }

    //递归删除节点
    //规定：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);
        }
        //左递归没有结束的话，就进行右递归进行删除
        if (this.right!=null){
            this.right.delNode(no);
        }
    }


}