数据结构与算法-二叉树

基本介绍

1. 树有很多种，每个节点最多只能有两个子节点的一种形式称为二叉树。

2. 二叉树的子节点分为左节点和右节点
   

3. 如果该二叉树的所有叶子节点都在最后一层，并且节点总数=2n-1,n为层数，则我们称为满二叉树

5. 如果该二叉树的所有叶子节点都在最后一层或者倒数第二层，而且最后一层的叶子节点在左边连续，倒数第二层的叶子节点在右边连续，我们称为完全二叉树

   

遍历

前序遍历：先输出父节点，再遍历左子树和右子树
中序遍历：先遍历左子树，再输出父节点，再遍历右子树
后序遍历：先遍历左子树，再遍历右子树，最后输出父节点

前序遍历：A , B , E , C , G
中序遍历：B , E , A , G , C
后序遍历：E , B , G , C , A

【代码实现】

public class BinaryTreeDemo {
    public static void main(String[] args) {
        //创建树
        BinaryTree binaryTree = new BinaryTree();
        //创建节点
        HeroNode root = new HeroNode(1,"A");
        HeroNode root2 = new HeroNode(2,"B");
        HeroNode root3 = new HeroNode(3,"C");
        HeroNode root4 = new HeroNode(4,"E");
        HeroNode root5 = new HeroNode(5,"G");
        //手动创建二叉树
        root.setLeft(root2);
        root.setRight(root3);
        root2.setRight(root4);
        root3.setLeft(root5);
        binaryTree.setRoot(root);//重要！！！ 将节点放到树中
        //测试
        System.out.println("前序遍历：");
        binaryTree.preOrder();
        System.out.println("中序遍历：");
        binaryTree.infixOrder();
        System.out.println("后序遍历：");
        binaryTree.postOrder();

    }
}

//创建一颗树
class BinaryTree{
    private HeroNode root;

    public void setRoot(HeroNode 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("二叉树为空");
        }
    }
}

//创建树节点
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 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 HeroNode getLeft() {
        return left;
    }

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

    public HeroNode getRight() {
        return right;
    }

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

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

    /**
     * 前序遍历
     * 先输出当前节点，
     * 如果左子节点不为空，则递归
     * 如果右子节点不为空，则递归
     */
    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();
        }
        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);//父节点
    }
}

【运行结果】

查找

【说明】
前序查找
1、先判断当前节点是否等于要查找的值
2、如果相等，则返回当前节点
3、如果不等，则判断当前节点的左子节点，如果不为空，则递归
4、如果左递归前序查找，找到节点，则返回，否继续判断，当前节点的右子节点是否为空，如果不为空，则继续向右递归查找

中序查找
1、判断当前节点的左子节点，如果不为空，则递归中序查找
2、如果找到则返回，如果没有，就和当前节点比较，是则返回当前节点，否则继续向右递归中序查找
3、如果右递归中序查找，找到则返回，否则返回null

后序查找
1、判断当前节点的左节点是否为空，如果不为空，则递归后序查找
2、如果找到，则返回，没有找到，就判断当前节点的右子节点是否为空，如果不为空，则右递归进行后序查找，如果找到，则返回
3、和当前节点进行比较，是则返回，否则返回null

【代码实现】

HeroNode类

/**
 * 前序查找
 * @param no 查找的no
 * @return 找到则返回节点，没有则返回null
 */
public HeroNode preOderSearch(int no){
    HeroNode res = null;//当前节点
    //比较
    if (this.no == no){
        return this;
    }
    //左递归查找
    if (this.left != null){
        res = this.left.preOderSearch(no);
    }
    if (res != null){//左子树找到
        return  res;
    }
    //右递归查找
    if (this.right != null){
        res = this.right.preOderSearch(no);
    }
    return res;
}

/**
 * 中序查询
 */
public HeroNode infixOrderSearch(int no){
    HeroNode res = null;//当前节点
    if (this.left != null){
        res = this.left.infixOrderSearch(no);
    }
    if (res != null){
        return res;
    }
    if (this.no == no){
        return this;
    }
    if (this.right != null){
        res = this.right.infixOrderSearch(no);

    }
    return res;
}
/**
 * 后序查询
 */
public HeroNode postOrderSearch(int no){
    HeroNode res = null;
    //左子节点
    if (this.left != null){
        res = this.left.postOrderSearch(no);
    }
    if (res != null){
        return res;
    }
    //右子树
    if (this.right != null){
        res = this.right.postOrderSearch(no);
    }
    if (res != null){
        return res;
    }
    //比较
    if (this.no == no){
        return this;
    }
    return res;
}

//BinaryTree类中  中序查找
public HeroNode infixOrderSearch(int no){
    if (root != null){
        return this.root.infixOrderSearch(no);
    }else {
        return null;
    }
}

//main方法
System.out.println("前序查询no=3：");
HeroNode res = binaryTree.preOrderSearch(3);

删除

要求：
如果删除的节点是叶子节点，则删除该节点
如果删除的节点是非叶子节点，则删除该子树.

思路：
1、二叉树是单向的，所有我们判断的是当前节点的子节点是否是需要删除的节点
2、如果当前节点的左子节点不为空，并且左节点就是要删除的节点，就将this.left = null。并且返回(结束递归删除)
3、如果当前节点的右子节点不为空，并且右子节点就是要删除的节点，就将this.right = null；并且返回
4、如果第2第3步没有删除节点，那么就向左子树进行递归
5、如果第4步没有删除节点，就向右子树进行递归

【代码实现】

//HeroNode类
/**
 *  简单删除节点
 * @param no
 */
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);
    }
}

//BinaryTree类中 
public void delNode(int no){
    if (root != null){
        //只有一个节点，判断是不是要删除的节点
        if (root.getNo() == no){
            root = null;
        }else {
            root.delNode(no);
        }
    }else {
        System.out.println("二叉树为空");
    }
}