leetcode 105. 从前序与中序遍历序列构造二叉树(哈希)

最新推荐文章于 2024-07-26 01:21:03 发布

matt__

最新推荐文章于 2024-07-26 01:21:03 发布

阅读量128

点赞数

分类专栏：树文章标签： leetcode 105. 从前序与中序遍历序列构造二叉树(哈希)

树专栏收录该内容

8 篇文章 0 订阅

订阅专栏

在这里插入图片描述

		/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<int,int>ans;
    TreeNode* dfs(vector<int>& pre, vector<int>& in,int pl,int pr,int il,int ir)//pl是指当前pl根节点在前序的范围，后面是在中序                                                                                                                  //的范围
    {
      if(pl>pr)return nullptr;//不存在范围说明为空
      int rec=pre[pl];//当前根的元素值
      int k=ans[rec];//根在中序遍历的下标
      int length=k-il;//根左孩子范围
      auto root=new TreeNode(rec);
      root->left=dfs(pre,in,pl+1,pl+length,il,k-1);//pl+length为pl左树的上界，k-1为中序的上界；
      root->right=dfs(pre,in,pl+1+length,pr,k+1,ir);
      return root;
      
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
     int n=inorder.size();
     for(int i=0;i<n;i++)ans[inorder[i]]=i;//把中序下标存入哈希表
     return dfs(preorder,inorder,0,n-1,0,n-1);
    }
};