给定两个整数数组 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);
}
};