一维数组构建二叉树【ACM模式】详解

最新推荐文章于 2024-02-25 19:58:52 发布

无敌颢克

最新推荐文章于 2024-02-25 19:58:52 发布

阅读量134

点赞数

分类专栏：算法入门文章标签： java 算法

本文链接：https://blog.csdn.net/weixin_59402226/article/details/134492745

版权

算法入门专栏收录该内容

2 篇文章 0 订阅

订阅专栏

节点类

  //节点类
    static class TreeNode{
        String val;
        TreeNode left;
        TreeNode right;
        public TreeNode(String val){
            this.val = val;
        }
    }

主函数

  //主函数
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        //接受输出
        String str = sc.nextLine();
        //空格分割为数组
        String[] arr  = str.split(" ");
        //构造二叉树
        TreeNode root = createBinaryTree(arr);
        //先序遍历并打印
        ergodicBinaryTree(root);
    }

构建二叉树【核心】

    //构建二叉树【核心】
    public static TreeNode createBinaryTree(String[] array){
        LinkedList<TreeNode> queue = new LinkedList<>();
        //创建根节点传入数组第一个元素
        TreeNode root = new TreeNode(array[0]);
        //根节点入队
        queue.offer(root);
        //指针
        int index = 1;
        while (!queue.isEmpty()) {
            TreeNode node = queue.pop();
            if (index < array.length){
                //创建左节点，默认为null
                TreeNode left = null;
                //如果数组中的值，不为空，就将它更新为左节点的新值
                if (array[index] != null) {
                    left = new TreeNode(array[index]);
                }
                //新的左节点入队
                queue.offer(left);
                //将左节点放入树中
                if(node != null){
                    node.left = left;
                }
                //索引后移，遍历数组下一个元素
                index++;
                //右节点与左节点步骤一致
                TreeNode right = null;
                if (array[index] != null) {
                    right = new TreeNode(array[index]);
                }
                queue.offer(right);
                if(node != null){
                    node.right = right;
                }
                index++;
            }
        }
        return root;
    }

先序遍历打印二叉树

 //遍历二叉树(先序)
    public static void ergodicBinaryTree(TreeNode root){
        if (root == null){
            return;
        }
        System.out.print(root.val+" ");
        ergodicBinaryTree(root.left);
        ergodicBinaryTree(root.right);
    }