首页 > 技术文章 > 剑指offer重构二叉树 给出二叉树的前序和后序重构二叉树

code-changeworld 2015-06-01 13:01 原文

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},
则重建二叉树并输出它的后序遍历序列。
(测试用例中,"树"的输出形式类似于树的层次遍历,没有节点的用#来代替)
先序遍历:先遍历根节点,再遍历左子树,再遍历右子树,对子树采用相对规则遍历。
后序遍历:先遍历左子树,然后遍历右子树,最后遍历根节点,对子树采用相同规则。
分析:
 
1:每棵前序遍历的第一个元素,一定是这个树的根节点
2:对每棵树遍历后序遍历序列,找到与根节点元素相等节点的位置pos,根据后序遍历的特点,根节点向左的序列为做子树的后序遍历序列{4, 7, 2},根节点向右的序列为右子树的后序遍历序列{5, 3,  8, 6}.
3:根据根节点的位置可以节点的位置,可以分别求出,左右子树节点数量,这样可以在前序遍历中确认,左右子树的的前序遍历分别为{2, 4, 7},{3, 5, 6, 8}
4:对左右子树重复1,2,3步。
5:递归返回条件,一个节点不存在子树时返回。
 
c++代码实现
 
class Solution {
    public:
        struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
            if (pre.empty() || in.empty()) return NULL;
            TreeNode *root = NULL;
            root = Constructcore(pre, in, 0, (int)pre.size()-1, 0, (int)in.size()-1);
            return root;
        }
        TreeNode *Constructcore(vector<int> pre, vector<int> in,
                                int startPre, int endPre,
                                int startIn, int endIn ){
            
            int rootValue = pre[startPre];
            TreeNode *root = new TreeNode(rootValue);
            if (startPre == endPre){
                if (startIn == endIn &&pre[startPre] == in[startIn])
                return root;
            }
            int pos;
            for (pos = startIn; pos <= endIn; pos++){
                if (pre[startPre] == in[pos]) break;
            }
            int leftlength = pos - startIn;
            int rightlength = endIn - pos;
            if (leftlength > 0){
                
                root->left = Constructcore(pre, in, startPre + 1,startPre + 1 +leftlength,
                                                    startIn,     pos - 1
                                           );
            }
            if (rightlength > 0){
                root->right = Constructcore(pre, in, startPre + 1 + leftlength, endPre,
                                                     pos + 1,            endIn
                
                );
            }
            return root;
        }
        
    };

 

 

推荐阅读