重建二叉树

重建二叉树

给定一个前序遍历序列和一个中序遍历序列

构建一棵二叉树

思路:

1. 前序的第一个节点一定是 根节点, 根据前序的第一个节点创建根节点
2. 找到根节点以后再根据根节点的值去中序序列中找根节点的位置
3. 根节点的左边就是左子树, 右边就是右子树
4. 然后递归构建左右子树

/**
 * Definition for binary tree
 * 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) {
        int pre_size = pre.size();
        int vin_size = vin.size();
        if(pre_size == 0 || vin_size == 0)
            return NULL;
        return help(pre, 0, pre_size-1, 
                    vin, 0, vin_size-1);
    }

    TreeNode* help(vector<int> pre, int pre_begin, int pre_end,  
                   vector<int> vin, int vin_begin, int vin_end)
    {
        if(pre_begin > pre_end || vin_begin > vin_end)
            return NULL;
        // 先根据前序遍历的第一个元素创建根节点
        TreeNode* root = new TreeNode(pre[pre_begin]);
        // 再根据中序遍历结果找到根节点在中序遍历序列中的位置
        for(int i = vin_begin; i <= vin_end; i++)
        {
            if(vin[i] == root->val)
            {
                root->left = help(pre, pre_begin + 1, pre_begin + i - vin_begin,
                                  vin, vin_begin, i-1);
                root->right = help(pre, pre_begin + i - vin_begin + 1, pre_end,
                                  vin, i+1, vin_end);
                break;
            }
        }
        return root;
    }
};