重建一个二叉树

给一个二叉树的先序和中序遍历,用数组表示,且这两个数组中没有重复的数字。让你重新建立一个二叉树。
其实这个题,我们能看出来这两个遍历之间的关系。先序遍历,数组的第一个就是根节点,第二个就是左子树的根节点(如果有左子树的话),以此类推。中序遍历,树的根节点在数组的中间,左边是左子树右边是右子树。
其实这个规律就是这样的,但是怎么用这个规律来重建一个二叉树呢?

我们想啊,我们开始找的得是整个树的根节点。根据这个根节点我们可以找到,左子树和右子树的根节点。然后依次往下找直到这个节点是个叶子节点就好了。那么我们就可以使用递归来借这个题。

代码:

struct TreeNode
{
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x): val(x),left(NULL), right(NULL) {}
};

class Solution
{
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
    {
        if(pre.size() ==0|| vin.size() == 0)//这两个表示遍历的数组不能为0
            return NULL;
        TreeNode* root = new TreeNode(pre[0]);//根据先序遍历找到根节点,每次先序遍历的第一个都是这个数的根节点
        int i=0;
        int len = vin.size();
        for(;i<len;++i)
        {
            if(vin[i] == pre[0])
                break;//找到在中序遍历中根节点所在的位置,那么前面的都是左子树后面的是右子树
        }
        vector<int> pre1(pre.begin()+1,pre.begin()+i+1);
        vector<int> pre2(pre.begin()+1+i,pre.end());
        vector<int> vin1(vin.begin(),vin.begin()+i);
        vector<int> vin2(vin.begin()+i+1,vin.end());
        root->left = reConstructBinaryTree(pre1,vin1);
        root->right = reConstructBinaryTree(pre2,vin2);
        return root;
    }
};
    原文作者:单倍体
    原文地址: https://www.jianshu.com/p/9d06929b2149
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞