重建二叉树【每日一练】

最新推荐文章于 2024-10-06 14:10:45 发布

KCode-2023

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量37

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/qq_43654232/article/details/130142324

版权

题目

题目描述：输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

1. 思想

前序遍历的第一个就是当前根节点，然后再在中序遍历中找出根节点位置，划分左右子树并递归建树。

1.1 优化

优化的关键在于如何快速在中序遍历中找出根节点位置
需要快速查找的情况，可以考虑使用哈希和二分搜索/有序结构（红黑树、B+树、跳表）。后者需要事先建立好数据结构，会带来额外的空间开销

2. 算法代码

 unordered_map<int, int> um;

   TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
       for(int i=0; i<inorder.size(); ++i)
           um[inorder[i]] = i;
       return buildTreeHelper_(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1);
   }

   TreeNode* buildTreeHelper_(vector<int>& preorder, vector<int>& inorder, int pleft, int pright, int ileft, int iright){
       if(pleft>pright || ileft>iright)
           return nullptr;
       if(pleft==pright && ileft==iright)
           return new TreeNode(preorder[pleft]);
       TreeNode* root = new TreeNode(preorder[pleft]);
       int rootIdx = um[preorder[pleft]];
       int leftSize = rootIdx-ileft;
       root->left = buildTreeHelper_(preorder, inorder, pleft+1, pleft+leftSize, ileft, rootIdx-1);
       root->right = buildTreeHelper_(preorder, inorder, pleft+leftSize+1, pright, rootIdx+1, iright);
       return root;
   }