• 重构二叉树


    重构二叉树

    这是剑指offer中关于二叉树重构的一道题。题目原型为:

    输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

    一、二叉树的数据结构


      做题之前,我们先熟悉下二叉树的数据结构。其一,定义:二叉树是一个连通的无环图,并且每一个顶点的度不大于3。有根二叉树还要满足根结点的度不大于2。有了根结点之后,每个顶点定义了唯一的父结点,和最多2个子结点。然而,没有足够的信息来区分左结点和右结点。如果不考虑连通性,允许图中有多个连通分量,这样的结构叫做森林。(定义来自百度百科)

      由定义可知,二叉树含有许多节点,而不同节点之间通过子父关系来连接。一般来说,二叉树的节点由结构体来定义,如下所示: 

    复制代码
    1 struct TreeNode {
    2     int val;
    3     TreeNode *left;
    4     TreeNode *right;
    5     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
    6     ~TreeNode() {
    7         cout << "TreeNode with value " << val << " has been destroyed." <<endl;
    8     }
    9 };
    复制代码

      该结构体定义了节点的int型变量(当然也可以是其它类型),以及指向左右孩子的指针。在TreeNode中,我们还定义了构造函数和析构函数,其实可有可无,对本题来说没有多大意义。

    二、二叉树的建立


      二叉树的建立过程,就是不断扩展节点的过程。建立根节点,由根节点建立左右子节点,再递归的建立子节点的子节点。我们下面看看代码的实现过程:

    复制代码
     1 void BiTree::createBiTree(struct TreeNode* &root) {
     2     int val;
     3     //cout << "Please input the tree node:" << endl;
     4     cin >> val;
     5     if (999 == val) {
     6         root = NULL;
     7     }
     8     else {
     9         //root = (struct TreeNode*) malloc(sizeof(TreeNode));
    10         root = new TreeNode(1);
    11         if (root == NULL)
    12             return;
    13         root->val = val;
    14 
    15         cout << "Please input the left child of " << val << ": ";
    16         createBiTree(root->left);
    17 
    18         cout << "Please input the right child of" << val << ": ";
    19         createBiTree(root->right);
    20     }
    21 }
    复制代码

      这里,我是把创建二叉树的方法createBiTree()作为了BiTree的成员函数(简单说明一下)。

      1、首先,我们看参数 struct TreeNode* &root, 这是结构体指针的引用。为什么要传递引用呢,就是希望函数体外的指针变量能指向函数体内所新建的根节点。若只是指针传递,那么函数体内新建的节点和函数体外的指针变量是没有指向关系的。

      2、然后就是新建节点,

      //root = (struct TreeNode*) malloc(sizeof(TreeNode));
         root = new TreeNode(1);
      这两条语句的效果相同。如果创建成功,那么递归调用本方法,创建后续节点。

    三、二叉树的遍历

      二叉树有三种遍历方法:先序,中序,后序。即父前(父-左-右),父中(左-父-右),父后(左-右-父)。
      


      1、先序遍历
    复制代码
    1 // 这里参数传递指针,或指针的引用都可以
    2 void BiTree::preOrder(struct TreeNode* &root) {
    3     if (root == NULL)
    4         return;
    5     cout << root->val << ' ';
    6     preTrav.push_back(root->val);
    7     preOrder(root->left);
    8     preOrder(root->right);
    9 }
    复制代码
      其中,preTraa.push_back(root->val);我只是把先序遍历的结果存在了vector 中,方便重构二叉树时作为输入。下同。

      2,、中序遍历

    复制代码
    1 void BiTree::inOrder(struct TreeNode* root) {
    2     if (root == NULL)
    3         return;
    4     inOrder(root->left);
    5     cout << root->val << ' ';
    6     inTrav.push_back(root->val);
    7     inOrder(root->right);
    8 }
    复制代码

      3、后序遍历,就不贴代码了。

      另外,值得说明一下的是。这里的遍历方法都是通过递归来实现的,还有遍历的非递归算法(日后再详细去探讨,这里先着重看这个题目,也不知日后是否会想起这个任务。。。囧)

    三、已知先序遍历,中序遍历,重构二叉树


      我们观察先序遍历和中序遍历的结果。先序遍历的第一个节点肯定是根节点,然后在中序遍历中找到此根节点,我们可以看到,中序遍历中根节点以左是二叉树的左子树,根节点以右是二叉树的右子树。然后,左右子女树又可以单独的看成一个独立的二叉树,然后再对其划分,如此递归,便可重构二叉树的结构。下面上代码:

    复制代码
     1 struct TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> in){
     2         //TreeNode* head = new TreeNode(pre[0]);
     3         /* 注意上面这一条语句不能放在函数体的第一句。
     4          * 因为此时还不能确定pre这个vector是否为空,如果不为空,那么访问pre[0]就是非法内存访问。此时若debug,会有
     5          * Program received signal SIGSEGV, Segmentation fault.的错误信息。其中,SIG是signal的缩写,SEGV是segmentation violation(段违例)的缩写。详见维基百科
     6          * 所以,此条语句放在return NULL;语句之后为宜,因为这个时候已经确定vector in不为空。但是好像没判断pre是否为空呢,这就可以了??
     7         */
     8         vector<int> left_pre, left_in, right_pre, right_in;
     9 
    10         int pos = 0;
    11         int length = in.size();
    12 
    13         if (length == 0)
    14             return NULL;
    15 
    16         TreeNode* head = new TreeNode(pre[0]);
    17 
    18         for (int i = 0; i < length; ++i) {
    19             if (pre[0] == in[i]) {
    20                 pos = i;
    21                 break;
    22             }
    23         }
    24 
    25         for (int i = 1; i < pos + 1; ++i) {
    26             left_pre.push_back(pre[i]);
    27             left_in.push_back(in[i-1]);
    28         }
    29 
    30         for (int i = pos + 1; i < length; ++i) {
    31             right_pre.push_back(pre[i]);
    32             right_in.push_back(in[i]);
    33         }
    34 
    35         head->left = reConstructBinaryTree(left_pre, left_in);
    36         head->right = reConstructBinaryTree(right_pre, right_in);
    37 
    38         return head;
    39 
    40     }
    复制代码

      其实,关于这个重构函数,笔者还有一个问题,就如函数中注释中所诉。还请广大博友解答下(新手博客,都没人看,好忧伤~_~!)

    四、重构验证


      首先,贴上验证程序。验证程序综合了上述建立二叉树,遍历二叉树以及重构二叉树的相关实现。

     View Code

      下面我们看看程序结果。

    从结果中我们可以看见,重构后二叉树的中序遍历和我们输入的二叉树的中序遍历相同。我们还可以输出重构后的先序遍历,来确定我们的重构是正确的。
    二叉树的重构暂且先探讨这。二叉树还有其它很多特性需要没我们去研究,择日再说。
      
     
    分类: 剑指offer
  • 相关阅读:
    verifycode验证码模版
    输出一个整数的每一位,如:123的每一位是1 , 2 , 3
    编写代码模拟三次密码输入的场景。 最多能输入三次密码,密码正确,提示“登录成功”,密码错误, 可以重新输 入,最多输入三次。三次均错,则提示退出程序
    获取一个数二进制序列中所有的偶数位和奇数位, 分别输出二进制序列
    一台客户端有三百个客户与三百个客户端有三百个客户对服务器施压,有什么区别?
    软件生存周期及其模型是什么?
    试述软件的概念和特点?软件复用的含义?构件包括哪些?
    Spring Cloud Stream与Spring Cloud Bus区别
    Spring的设计模式
    JavaScript的事件循环机制总结 eventLoop
  • 原文地址:https://www.cnblogs.com/Leo_wl/p/5721893.html
Copyright © 2020-2023  润新知