40-二叉树练习-牛客网KY11二叉树遍历

题目

编一个程序，读入用户输入的一串先序遍历字符串，根据此字符串建立一个二叉树（以指针方式存储）。 例如如下的先序遍历字符串： ABC##DE#G##F### 其中“#”表示的是空格，空格字符代表空树。建立起此二叉树以后，再对二叉树进行中序遍历，输出遍历结果。

输入描述：

输入包括1行字符串，长度不超过100。

输出描述：

可能有多组测试数据，对于每组数据， 输出将输入字符串建立二叉树后中序遍历的序列，每个字符后面都有一个空格。 每个输出结果占一行。

示例1

输入：abc##de#g##f###

输出：c b e g d f a

---

思路

---

代码

import java.util.Scanner;

public class Main {
    private static class TreeNode {
        char val;
        TreeNode left;
        TreeNode right;
        public TreeNode(char val) {
            this.val = val;
        }
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        //获取用户多组输入
        while (scanner.hasNext()) {
            //字符串形式的先序二叉树结果
            String line = scanner.next();
            //str -> TreeNode
            TreeNode root = build(line);
            //中序遍历二叉树，按照格式打印节点值
            inOrder(root);
            //每个结果占一行
            System.out.println();
        }
    }

    /**
     * 中序遍历打印二叉树
     * @param root
     */
    private static void inOrder(TreeNode root) {
        if (root == null) {
            return;
        }
        inOrder(root.left);
        System.out.print(root.val + " ");
        inOrder(root.right);
    }

    /**
     * 将字符串还原为二叉树
     * @param line
     * @return
     */
    private static TreeNode build(String line) {
        return preOrderBuild(line);
    }

    //当前处理到哪个字符了
    static int index = 0;

    /**
     * 根据先序遍历结果，将字符串还原为二叉树，返回构建后二叉树的根节点
     * @param line
     * @return
     */
    private static TreeNode preOrderBuild(String line) {
        //取得当前需要处理的字符
        char c = line.charAt(index);
        if (c == '#') {
            return  null;
        }
        //根节点
        TreeNode root = new TreeNode(c);
        //继续处理下一个字符
        index++;
        //继续处理左子树
        root.left = preOrderBuild(line);
        index++;
        root.right = preOrderBuild(line);
        return root;
    }
}