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);
}
};//一定要使用引用否则会超时

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