九度 1184：二叉树遍历

题目描述：

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

 

思路

1. 想了半天, 写不出递归函数, 倒是基于堆栈的循环方法简单的多

 

代码

#include <iostream> #include <stdio.h> #include <deque> #include <set> #include <cstring>
using namespace std; class Node { public: char data; Node *left, *right; Node(char _data):data(_data) { left = right = NULL; } }; char seq[200]; int n; Node* buildTree() { if(n == 1) { if(seq[0] == '#') return NULL; return new Node(seq[0]); } set<Node*> lchild; deque<Node*> stack; Node *root = new Node(seq[0]); stack.push_back(root); for(int i = 1; i < n; i ++) { if(seq[i] == '#') { Node *lastone = stack.back(); if(lchild.count(lastone)) { lastone->right = NULL; stack.pop_back(); }else{ lchild.insert(lastone); lastone->left = NULL; } }else{ Node *lastone = stack.back(); Node *newNode = new Node(seq[i]); if(lchild.count(lastone)) { lastone->right = newNode; stack.pop_back(); stack.push_back(newNode);// WA }else{ lchild.insert(lastone); lastone->left = newNode; stack.push_back(newNode); } } } return root; } void inOrder(Node *root) { if(root == NULL) return; inOrder(root->left); printf("%c ", root->data); inOrder(root->right); } int main() { while(scanf("%s", seq) != EOF) { n = strlen(seq); Node *root = buildTree(); inOrder(root); printf("\n"); } return 0; }