给定两个整数数组 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]
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:unordered_map<int,int> hash;//哈希表用于存放中序序列值的下标,方便查找TreeNode* build(vector<int>& preorder,vector<int>& inorder,int pl,int pr,int il,int ir){if(pl>pr||il>ir){return nullptr;}TreeNode* node=new TreeNode(preorder[pl]);//前序序列的第一个节点是此轮的根节点int split=hash[preorder[pl]];//根节点在中序序列中的下标int leftnums=split-il;//左子树的节点数//左孩子前中序序列边界不同:根节点+[序列长度]or[序列长度]+根节点node->left=build(preorder,inorder,pl+1,pl+leftnums,il,il+leftnums-1);//右孩子前中序序列的边界是一样的node->right=build(preorder,inorder,pl+leftnums+1,pr,il+leftnums+1,ir);return node;}TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {//时间复杂度O(N),空间复杂度O(N)int n=preorder.size();for(int i=0;i<n;i++){hash[inorder[i]]=i;}//初始化哈希表//pl:前序序列左边界 pr:前序序列右边界 il:中序序列左边界 ir:中序序列右边界return build(preorder,inorder,0,n-1,0,n-1);}
};