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

主要是二叉树的前序，中序和后序的查找遍历

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
 * @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);
        }
    }
}

//定义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;
        }
    }
}


//先创建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;
    }

}