给定两个整数数组 preorder
和 inorder
,其中 preorder
是二叉树的先序遍历, inorder
是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] 输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1] 输出: [-1]
代码如下:
class Solution {
public:unordered_map<int,int> index;//定义一个哈希表,存放中序遍历的节点值对应的下标TreeNode* myBuildTree(vector<int>& preorder,vector<int>& inorder,int preleft,int preright,int inleft,int inright)//定义中序遍历与前序遍历的下标{if(preleft>preright||inleft>inright){return nullptr;}int preroot=preleft;//前序遍历中的第一个节点就是根节点int inroot=index[preorder[preroot]];//在中序遍历中定义根节点TreeNode* root=new TreeNode(preorder[preroot]);//创建二叉树的根节点int leftsize=inroot-inleft;//计算左子树中的节点数//递归的进行左子树,连接到根节点//在先序遍历中,左子树preleft+1到preleft+leftsize对应中序遍历中inleft到inroot-1root->left=myBuildTree(preorder,inorder,preleft+1,preleft+leftsize,inleft,inroot-1);//递归的进行右子树//在先序遍历中,右子树preleft+leftsize+1到preright对应中序遍历中inroot+1到inrightroot->right=myBuildTree(preorder,inorder,preleft+leftsize+1,preright,inroot+1,inright);return root;}TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {int n=preorder.size();for(int i=0;i<n;i++){index[inorder[i]]=i;}return myBuildTree(preorder,inorder,0,n-1,0,n-1);}
};