leetcode 重建二叉树

kotomi_du

于 2018-08-24 11:43:53 发布

阅读量516

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/github_38818603/article/details/82015404

算法专栏收录该内容

57 篇文章

订阅专栏

本文深入探讨了基于前序和中序遍历重构二叉树的递归算法。通过详细解析递归过程中的关键步骤，如返回条件、索引更新及子树构造，为读者提供了清晰的算法实现思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

key:

1. 递归的返回条件：ps > pe;

2. 左右子树的索引更新：其中index是根据前序遍历在中序遍历中找到的值

preorder: (ps+1, ps+index-is); (ps+index-is+1, pe)

inorder: (is, index-1); (index+1, ie)

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {       
          if(pre.size() == 0 || vin.size() == 0 || pre.size() != vin.size()){
              return NULL;
          }
        int len =pre.size();
        return Construct(pre, 0 , len -1, vin , 0, len-1);
    }
    
     TreeNode* Construct(vector<int> pre, int ps, int pe, vector<int> vin, int vs, int ve){
         if(ps > pe)  return NULL;
         int val = pre[ps];
         int index = vs;
         while(index <= ve){
             if(vin[index] == val) break;
             index++;
         }
         if(index > ve) return NULL;
         TreeNode* node = new TreeNode(val);
     
         node->left= Construct(pre,ps+1, ps+index-vs, vin, vs, index-1);         
         node->right = Construct(pre,ps+index-vs+1, pe, vin, index+1, ve);
         return node; 
     }
    
};