java实现二叉树

概述

实现

定义树中节点类

package com.lovehena.datastructure;

import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.NoArgsConstructor;

/*
*   java实现一颗二叉树
*       二叉树的特点：
*           只有一个根节点。
*           每个节点左侧的节点值都是小于该节点的，右侧的节点值都是大于该节点的。
*           每个节点的子节点最多为2个（≤2,即 0或1或2）
*   如二叉树：
*          1
*       2      3
*     4   5  6   7
* */
@Getter
@AllArgsConstructor
@NoArgsConstructor
public class BinaryTreeNode {

    private int value; // 当前节点的值
    private BinaryTreeNode left; // 当前节点的左子节点
    private BinaryTreeNode right; // 当前节点的右子节点

    /**
     *
     * @param value 用于构造没有左子节点、右子节点的节点
     */
    public BinaryTreeNode(int value) {
        this.value = value;
    }


}

二叉树的层序遍历

层序遍历定义

按层次从上到下、从左到右依次访问节点

实现

	/**
     * 二叉树的层序遍历（levelOrderTraversal）：
     *  按层次从上到下、从左到右依次访问节点
     * 如二叉树：
     *          1
     *       2      3
     *     4   5  6   7
     * 的层序遍历为：1 2 3 4 5 6 7
     *
     * @param leftNode 左子节点
     * @param leftNode 右子节点
     * @param sb       收集树中节点值
     */

    private static void levelOrderTraversal(BinaryTreeNode leftNode, BinaryTreeNode rightNode, StringBuilder sb) {
        // 从左到右依次访问节点 意思就是 先遍历左节点的值 再遍历右节点的
        // 每个左节点、右节点的左子节点、右子节点遍历时也是这个步骤
        if (leftNode != null) {
            sb.append(leftNode.getValue()).append(" "); // 先遍历左节点的值
        }
        if (rightNode != null) {
            sb.append(rightNode.getValue()).append(" "); // 再遍历右节点的值
        }
        if (leftNode != null) { // 每个左节点的左子节点、右子节点遍历时也是这个步骤

            levelOrderTraversal(leftNode.getLeft(), leftNode.getRight(), sb);
        }
        if (rightNode != null) { // 每个右节点的左子节点、右子节点遍历时也是这个步骤
            levelOrderTraversal(rightNode.getLeft(), rightNode.getRight(), sb);
        }
    }

构造树 && 层序遍历二叉树

	public static void main(String[] args) {
        // 构造根节点 根节点下挂左子节点、右子节点
        // 左子节点、右子节点下又分别挂了自己的左子节点、右子节点 以此类推
        // 一棵树就形成了
        BinaryTreeNode rootNode = new BinaryTreeNode(
                1,// 根节点的值
                new BinaryTreeNode( // 根节点的左子节点
                        2,
                        new BinaryTreeNode(4),
                        new BinaryTreeNode(5)
                ),
                new BinaryTreeNode(// 根节点的右子节点
                        3,
                        new BinaryTreeNode(6),
                        new BinaryTreeNode(7)
                )
        );
        StringBuilder sb = new StringBuilder();
        // 从根节点遍历 没有和根节点同级的节点所以 传rightNode为null
        // 等同于这个调用 printTree(null, rootNode, sb);
        levelOrderTraversal(rootNode, null, sb);
        log.info("二叉树的层序遍历序列为：{}", sb.toString());

    }

测试用例输出

扩展

二叉树的中序遍历、前序遍历、后序遍历如何实现？

最后

好了，如果对你有帮助的话，欢迎点个免费的赞哦。