题目:给定二叉树的先序和中序遍历,构建这个二叉树。例如:先序遍历结果是:abdcef,中序遍历的结果是:dbaecf,如何构建二叉树
分析:先序遍历的结果中的每一个节点,将中序遍历的结果分为左右子树两部分,递归构建二叉树,就可以完成树得构建。
程序如下:
#include <stdio.h>
#include <string>
class TreeNode {
public:
TreeNode() : left_(NULL), right_(NULL) {}
TreeNode(char value) : value_(value), left_(NULL), right_(NULL) {}
char value_;
TreeNode* left_;
TreeNode* right_;
};
void BuildTree(const std::string& pre_order, size_t& offset, const std::string& in_order, int start, int end, TreeNode* &parent) {
if (offset < pre_order.size() && end >= start) {
parent = new TreeNode(pre_order[offset]);
char tmp = pre_order[offset];
offset++;
int pivot = start;
while (pivot <= end) {
if (in_order[pivot] != tmp) {
pivot++;
} else {
break;
}
}
BuildTree(pre_order, offset, in_order, start, pivot - 1, parent->left_);
BuildTree(pre_order, offset, in_order, pivot + 1, end, parent->right_);
}
}
void PreOrderVisit(TreeNode* current) {
if (current) {
printf("%c ", current->value_);
PreOrderVisit(current->left_);
PreOrderVisit(current->right_);
}
}
void InOrderVisit(TreeNode* current) {
if (current) {
InOrderVisit(current->left_);
printf("%c ", current->value_);
InOrderVisit(current->right_);
}
}
int main(int argc, char** argv) {
std::string pre_order = "abdcef";
std::string in_order = "dbaecf";
TreeNode* tree;
size_t offset = 0;
BuildTree(pre_order, offset, in_order, 0, in_order.size() - 1, tree);
PreOrderVisit(tree);
printf("\n");
InOrderVisit(tree);
}
注意:刚开始将start和end类型设置为size_t类型,程序执行总是不对,以至于快颠覆了我对递归的理解,不过跟踪调试的过程,更好的理解了递归的处理过程。因为start或end可能取值-1,而size_t是无符号的,导致判断出错,这个问题花费了一些时间来检查,下次一定注意,涉及到递归偏移量的问题,尽量使用有符号数,因为中间的过程可能出现负数。
参考文献:
编程之美3.9