根据前序遍历构建二叉树
题目描述
编一个程序,读入用户输入的一串先序遍历字符串,根据此字符串建立一个二叉树(以指针方式存储)。 例如如下的先序遍历字符串: ABC##DE#G##F### 其中“#”表示的是空格,空格字符代表空树。建立起此二叉树以后,再对二叉树进行中序遍历,输出遍历结果。
输入描述:
输入包括1行字符串,长度不超过100。
输出描述:
可能有多组测试数据,对于每组数据, 输出将输入字符串建立二叉树后中序遍历的序列,每个字符后面都有一个空格。 每个输出结果占一行。
示例
原题OJ链接
https://www.nowcoder.com/practice/4b91205483694f449f94c179883c1fef?tpId=60&&tqId=29483&rp=1&ru=/activity/oj&qru=/ta/tsing-kaoyan/question-ranking
解答
import java.util.Scanner;
// 注意类名必须为 Main, 不要有任何 package xxx 信息
class TreeNode{
//创建一个树结点的类,这个类最后不要构建在Main类了里面,构建在里面是内部类
char val;
TreeNode left;
TreeNode right;
public TreeNode(char ch){
this.val = ch;
this.left = null;
this.right = null;
}
}
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
// 注意 hasNext 和 hasNextLine 的区别
while (in.hasNextLine()) {
// 注意 while 处理多个 case
String str = in.nextLine();
TreeNode root = creatTree(str);
midOrder(root);
}
}
public static int i = 0;//必须是静态的,不然在creatTree这个静态方法里面没法用
//构建二叉树
public static TreeNode creatTree(String str){
TreeNode root = null;
if(str.charAt(i) != '#'){
//只要值不是#,就证明是结点,所以先把结点创建出来
root = new TreeNode(str.charAt(i));
i++;
root.left = creatTree(str);
root.right = creatTree(str);
}
else{
i++;//本题是不需要判断i是否越界的,因为只有给定的str是合法的前驱遍历,那么i就不会无限增加,因为有递归次数做保证
}
return root;
}
//中序遍历
public static void midOrder(TreeNode root){
if(root == null){
return;
}
midOrder(root.left);
System.out.print(root.val + " ");
midOrder(root.right);
}
}