根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
方法:使用递归。
具体是:前序遍历的第一个数是根节点。在中序遍历中找到根节点,然后根节点之前是左子树,(对应到前序遍历,左子树是从第一个节点之后,长度是中序遍历左子树的长度),根节点之后是右子树(对应到前序遍历,右子树是左子树的剩下的)。然后递归。
递归的终止条件:前序遍历索引大于前序数组的长度,或者中序遍历索引大于中序数组的长度。
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
return buildbinarytree(preorder, inorder, 0, 0,
inorder.size()-1);
}
TreeNode* buildbinarytree(vector<int>& preorder, vector<int>& inorder, int preindex, int instart, int inend ){
if( preindex > preorder.size()-1 || instart > inend)
return nullptr;
TreeNode* node = new TreeNode(preorder[preindex]);
int index = instart;
while( index <= inend){
if(preorder[preindex] == inorder[index]) break;
index++;
}
node->left = buildbinarytree(preorder,inorder, preindex + 1, instart, index-1);
node->right = buildbinarytree(preorder,inorder, preindex + (index-instart+1), index+1, inend);
return node;
}
};