【LeetCode】105. Construct Binary Tree from Preorder and Inorder Traversal

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

For example, given

preorder = [3,9,20,15,7]
inorder = [9,3,15,20,7]

Return the following binary tree:

    3
   / \
  9  20
    /  \
   15   7

      题目要求就是根据前序遍历和中序遍历的结果来还原二叉树。

      首先应该明白前序遍历和中序遍历的遍历顺序：

        1. 前序遍历（先根再左后右）；

        2. 中序遍历（先左再根后右）。

      我们首先找到根节点。对前序来说，这个二叉树的顺序组成是：根节点{左子树}{右子树}。

      对中序遍历来说，二叉树顺序组成是：{左子树}根节点{右子树}。

      所以先序的第一个值就是根节点，从而可以将二叉树的左右子树划分出来，然后将左右子树再重复以上步骤，就可以恢复出原来的二叉树。

/**
 * 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) {
        vector<int> pre_l,pre_r,in_l,in_r;
        TreeNode* root=NULL;
        
        if(!preorder.empty() && !inorder.empty()){
            root = new TreeNode(preorder[0]);
            int index=0;
            for(int i=0;i<inorder.size();i++){//找到根节点
                if(inorder[i]==preorder[0])
                    index=i;
            }
            for(int i=0;i<index;i++){//找到左子树
                pre_l.push_back(preorder[i+1]);
                in_l.push_back(inorder[i]);
            }
            for(int i=index+1;i<inorder.size();i++){//找到右子树
                pre_r.push_back(preorder[i]);
                in_r.push_back(inorder[i]);
            }
            //重复以上步骤
            root->left=buildTree(pre_l,in_l);
            root->right=buildTree(pre_r,in_r);
        }
        return root;
        
    }
};

下面是参考链接。点我