二叉树 | 建立二叉树

问题：从前序与中序遍历序列构造二叉树

问题链接

解题思路

前序遍历：根、左、右
中序遍历：左、根、右
前序遍历向量中的第一个元素就是根节点，我们通过根节点就可以将中序遍历分割成3部分，左、根、右，递归建立即可

C++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        //如果没节点了返回NULL
        if(preorder.size() == 0 || inorder.size() == 0) return NULL;
        vector<int>left, right;
        //将inorder中在根节点pre[0]左边的数归入left中
        int i = 0, size = inorder.size();
        while(i < size && inorder[i] != preorder[0]){
            left.push_back(inorder[i]);
            i++;
        }
        //将inorder中在根节点pre[0]右边的数归入right中
        i++;
        while(i < size){
            right.push_back(inorder[i]);
            i++;
        }
        TreeNode *root = new TreeNode(preorder[0]);//建立根节点
        preorder.erase(preorder.begin());//删除第一个元素
        root->left = buildTree(preorder, left);//递归建立左子树
        root->right = buildTree(preorder, right);//递归建立右子树
        return root;//返回根节点
    }
};

问题：从中序与后序遍历序列构造二叉树

问题链接

解题思路

1、通过后序遍历的最后一个节点获得根节点的值
2、根据根节点值将中序遍历序列进行分割，左右子树分别存放在left和right中
3、递归建立右子树，然后递归建立左子树

C++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        //如果没节点了返回NULL
        if(inorder.size() == 0 || postorder.size() == 0) return NULL;
        //根据根节点值将中序遍历序列进行分割，左右子树分别存放在left和right中
        vector<int>left, right;
        int rootVal = postorder.back();//根节点值
        //将inorder中在根节点rootVal左边的数归入left中
        int i = 0, size = inorder.size();
        while(i < size && inorder[i] != rootVal){
            left.push_back(inorder[i]);
            i++;
        }
        //将inorder中在根节点rootVal右边的数归入right中
        i++;
        while(i < size){
            right.push_back(inorder[i]);
            i++;
        }
        postorder.pop_back();//在后序遍历序列中删除根节点
        TreeNode *root = new TreeNode(rootVal);//建立根节点
        root->right = buildTree(right, postorder);//注意：先递归建立右子树
        root->left = buildTree(left, postorder);//递归建立左子树
        return root;//返回根节点
    }
};