一,题目描述
编一个程序,读入用户输入的一串先序遍历字符串,根据此字符串建立一个二叉树(以指针方式存储)。
例如如下的先序遍历字符串: ABC##DE#G##F### 其中“#”表示的是空格,空格字符代表空树。建立起此二叉树以后,再对二叉树进行中序遍历,输出遍历结果。
二,代码实现
1,思路
解决问题的关键在于怎样根据前序遍历的结果构造出这棵树,我们都知道前序遍历的方式为“根左右”,图示:
为了构造出这棵树我们采用递归的思想(流程看代码注释)
2,代码
import java.util.Scanner;
class TreeNode{ //树的结点的类
public char val;
TreeNode left;
TreeNode right;
public TreeNode(char val) {
this.val = val;
}
}
public class BuildTreeDemo {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()){
String str = scanner.next();
TreeNode root = build(str); //写一个方法根据输入先构建一棵树
inOrder(root); //前序遍历这颗树
System.out.println();
}
}
private static int index = 0;
private static TreeNode build(String str){
index = 0;//使用一个变量来记录一下
//index 是为了后面递归过程中能知道访问到哪个节点了
//通过下面的方法辅助完成
return creatTreePreOrder(str);
}
private static TreeNode creatTreePreOrder(String str) {
char cur = str.charAt(index);
if (cur == '#') {
return null;
}
TreeNode root = new TreeNode(cur); //当前字符不是'#'就创建一个新的结点
index++; //准备处理下一个节点, 下一个结点开始就是root的左子树的先序遍历结点
root.left = creatTreePreOrder(str);
index++; //准备处理下一个节点, 下一个结点开始就是root的右子树的先序遍历结点
root.right = creatTreePreOrder(str);
return root;
}
private static void inOrder(TreeNode root){
if(root == null){
return;
}
inOrder(root.left);
System.out.print(root.val + " ");
inOrder(root.right);
}
}