-
题目描述:
-
编一个程序,读入用户输入的一串先序遍历字符串,根据此字符串建立一个二叉树(以指针方式存储)。
例如如下的先序遍历字符串:
ABC##DE#G##F###
其中“#”表示的是空格,空格字符代表空树。建立起此二叉树以后,再对二叉树进行中序遍历,输出遍历结果。
-
输入:
-
输入包括1行字符串,长度不超过100。
-
输出:
-
可能有多组测试数据,对于每组数据,
输出将输入字符串建立二叉树后中序遍历的序列,每个字符后面都有一个空格。
每个输出结果占一行。
-
样例输入:
-
abc##de#g##f###
-
样例输出:
-
c b e g d f a
这题主要是要根据先序遍历构建二叉树, 这个代码写出来非常难看。 待优化, 下面是代码
import java.io.BufferedInputStream;
import java.io.IOException;
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static void main(String[] args) throws IOException {
Scanner cin = new Scanner(new BufferedInputStream(System.in));
while(cin.hasNext()) {
String string = cin.nextLine().trim();
process(string);
}
cin.close();
}
static void process(String string){
Node parent = null;
Node root = null;
Node node = null;
boolean left = true;
OUTER: for(int i=0; i<string.length(); i++) {
char ch = string.charAt(i);
if(ch == '#') {
if(left) {
left = false;
} else {
while(true) {
parent = parent.parent;
if(parent == null) {
break OUTER;
}
if(parent.right == null) {
break;
}
}
}
continue;
}
node = new Node(ch);
if(root == null) {
root = node;
}
if(parent != null) {
if(left) {
parent.left = node;
} else {
parent.right = node;
left = true;
}
node.parent = parent;
}
parent = node;
}
Tree tree = new Tree();
tree.root = root;
tree.middleTraversal();
}
}
class Tree {
Node root;
void middleTraversal() {
if(this.root == null) {
return;
}
StringBuffer buffer = new StringBuffer();
Stack<Node> stack = new Stack<Node>();
Node node = this.root;
while(node != null || !stack.isEmpty()) {
while(node != null) {
stack.push(node);
node = node.left;
}
Node pop = stack.pop();
buffer.append(pop.key).append(" ");
node = pop.right;
}
System.out.println(buffer.toString());
}
}
class Node {
char key;
Node left;
Node right;
Node parent;
Node(char key) {
this.key = key;
}
}