您的位置:首页 > Web前端

剑指offer之重建二叉树

2018-03-26 18:37 603 查看
问题:输入某二叉树的前序遍历和中序遍历的结果,请重新建出该二叉树。假设输入的前序遍历和中序遍历的结果中都是不含重复的数字。例如:输入的前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重新建立二叉树并返回。
知识点:
前序遍历:父节点--左子节点--右子节点 (又叫先序遍历,先根遍历);
中序遍历:左子节点--父节点--右子节点 (又叫中根遍历);
解决思路:递归,根据前序遍历的第一个数字创建根节点;在中序遍历序列中找到根节点的位置,确定左、右子树节点数量;然后递归构建左、右子树。
具体代码:(C++)
typedef struct TreeNode{
    int val;

    TreeNode* left;

    TreeNode* right;

    TreeNode(int x):val(x),left(nullptr),right(nullptr){}

};
class Solution
{
public:
    TreeNode* reConstructBinaryTree(vector<int>pre,vector<int>vin)

    {

        if(pre.empty() || vin.empty() || pre.size() != vin.size())

                return nullptr;

         vector<int> left_pre,right_pre,left_vin,right_vin;

         TreeNode* node = new TreeNode(pre[0]);

         int left_length = 0;

        while(pre[0] != vin[left_length] && left_length < pre.size())

                ++left_length;      //左子树的长度

        for(int i=0;i<left_length;i++)

        {

                left_pre.push_back(pre[i+1]);       //左子树的前序序列 ,pre[0]为根节点

                left_vin.push_back(vin[i]);            //左子树的中序序列

        }

        for(int i=left_length + 1; i<pre.size();++i)

        {
                right_pre.push_back(pre[i]);        //右子树的前序序列

                right_vin.push_back(vin[i]);        //右子树的中序序列

         }

         node->left = reConstructBinaryTree(left_pre,left_vin); //递归创建左子树

          node->right = reConstructBinaryTree(right_pre,right_vin); //递归创建右子树

        return node;

    }

};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: