LeetCode105—Construct Binary Tree from Preorder and Inorder Traversal

LeetCode105—Construct Binary Tree from Preorder and Inorder Traversal

原题

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

给出树的先序和中序遍历，构建二叉树。

分析

这题以前数据结构考试题中用，先序遍历可以可以找出树的根，中序遍历可以找出树的左右子树，如此递归下去，可以构建树。
根据此：
1. 在先序遍历中找到根
2. 在中序遍历中找到根对应的位置，即可把中序分为左右两子树
3. 计算在先序遍历中，分别计算左右子树中根的位置
4. 递归左右子树

关于第2点，我们需要提前建立一个map来找出根和中序遍历中索引的对应关系。

代码

class Solution {
private:
    TreeNode* helper(vector<int>&preorder, vector<int>&inorder, map<int, int>&index, int pstart, int pend, int istart, int iend)
    {
        if (istart > iend)
            return NULL;
        int rootval = preorder[pstart];//根
        int rootindex = index[rootval];//根在中序中的索引
        TreeNode* root = new TreeNode(rootval);
        root->left = helper(preorder, inorder, index, pstart + 1, pstart + rootindex - istart, istart, rootindex - 1);//递归左子树
        root->right = helper(preorder, inorder, index, pstart + rootindex - istart + 1, pend, rootindex + 1, iend);//递归右子树
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        map<int, int>index;
        for (int i = 0; i < inorder.size(); i++)
        {
            index[inorder[i]] = i;//建立值与索引的关系
        }
        return helper(preorder, inorder, index,0, preorder.size() - 1, 0, inorder.size() - 1);
    }
};//一定要使用引用否则会超时

这里传参数时一定要使用引用否则会超时。