重建二叉树

题目:给定二叉树的先序和中序遍历,构建这个二叉树。例如:先序遍历结果是: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

发布了82 篇原创文章 · 获赞 3 · 访问量 26万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章