根据前序遍历和中序遍历树构造二叉树.
注意事项
你可以假设树中不存在相同数值的节点
样例
给出中序遍历:[1,2,3]
和前序遍历:[2,1,3]
. 返回如下的树:
2
/
1 3
思路:若前序遍历为空或者中序遍历为空则返回空;
创建根节点,前序遍历第一个节点是个根节点,在中序遍历中找到根节点相应的位置,可以分别得到左子树和右子树的前序和中序遍历。重建左右子树;
使用辅助空间,存放被分割开的前序和中序遍历的序列。
/** * Definition of TreeNode: * class TreeNode { * public: * int val; * TreeNode *left, *right; * TreeNode(int val) { * this->val = val; * this->left = this->right = NULL; * } * } */ class Solution { /** *@param preorder : A list of integers that preorder traversal of a tree *@param inorder : A list of integers that inorder traversal of a tree *@return : Root of a tree */ /* 思路:若前序遍历为空或者中序遍历为空则返回空; 创建根节点,前序遍历第一个节点是个根节点,在中序遍历中找到根节点相应的位置, 可以分别得到左子树和右子树的前序和中序遍历。重建左右子树; 使用辅助空间,存放被分割开的前序和中序遍历的序列。 */ public: TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) { // write your code here if(preorder.size()==0||inorder.size()==0){ return NULL; } if(preorder.size()!=inorder.size()){ return NULL; } vector<int> preorder_l,preorder_r,inorder_l,inorder_r; int root_index=-1; int len=inorder.size(); TreeNode* root=new TreeNode(preorder[0]); // 在中序队列中找出根节点位置 for(int i=0;i<len;i++){ if(preorder[0]==inorder[i]){ root_index=i; break; } } // 左右子树的前序、中序队列 for(int i=0; i<root_index; i++) { preorder_l.push_back(preorder[i+1]); inorder_l.push_back(inorder[i]); } for(int i=root_index+1; i<inorder.size(); i++) { preorder_r.push_back(preorder[i]); inorder_r.push_back(inorder[i]); } root->left=buildTree(preorder_l, inorder_l); root->right = buildTree(preorder_r, inorder_r); return root; } };