关于程序员:算法题重建二叉树

6

要求

工夫限度：1 秒 空间限度：32768K

题目形容

输出某二叉树的前序遍历和中序遍历的后果，请重建出该二叉树。假如输出的前序遍历和中序遍历的后果中都 不含反复的数字 。例如输出前序遍历序列{1,2,4,7,3,5,6,8} 和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回

解题思路

解题的前提：两个序列中必须有一个中序序列

前（后）序序列用于提供根节点，根节点就是序列第一个元素（最初一个元素）

中序序列用于确定以后根节点的左右子树有哪些节点

如下图所示：

确定了左右子树有哪些元素之后就持续递归

代码

/*
Definition for binary tree
struct TreeNode
{
  int val;
  TreeNode *left;
  TreeNode *right;
  TreeNode(int x) : val(x), left(NULL), right(NULL) {}};
*/
class Solution
{
public:
  TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
  {int psize=pre.size();
    int vsize=vin.size();
      
    if(psize==0 || vsize==0 || psize != vsize)
      return NULL;
      
    return reBuildBinaryTreeCore(pre, 0, psize-1, vin, 0, vsize-1);
  }
  
  TreeNode* reBuildBinaryTreeCore(vector<int>& pre, int startIdxOfP, int endIdxOfP,
                                  vector<int>& vin, int startIdxOfV, int endIdxOfV)
  {int valOfRoot = pre[startIdxOfP];
    TreeNode* root = new TreeNode(valOfRoot);
    
    if(startIdxOfP == endIdxOfP)
    {if(startIdxOfV == endIdxOfV)
        return root;
      else
      {
        cout << "有效输出" << endl;
      abort();}
    }
    else
    {
      int idxOfRootInV = 0;
      for(idxOfRootInV = startIdxOfV;idxOfRootInV <= endIdxOfV && 
          vin[idxOfRootInV] != valOfRoot;++idxOfRootInV)
      { }
      if(vin[idxOfRootInV] != valOfRoot)
      {
        cout << "有效输出" << endl;
        abort();}
        
      int leftLength = idxOfRootInV - startIdxOfV;
      // 左
      if(startIdxOfV <= idxOfRootInV - 1)
        root->left = reBuildBinaryTreeCore(pre,
                                           startIdxOfP + 1,
                                           startIdxOfP + leftLength,
                                           vin,
                                           startIdxOfV,
                                           idxOfRootInV - 1);
      // 右
      if(idxOfRootInV + 1 <= endIdxOfV)
        root->right = reBuildBinaryTreeCore(pre,
                                            startIdxOfP + 1 + leftLength,
                                            endIdxOfP,
                                            vin,
                                            idxOfRootInV+1,
                                            endIdxOfV);
    }
    
    return root;
  }
};

小弟才浅，如果本篇文章有任何谬误和倡议，欢送大家留言

感激大家的点赞、珍藏

微信搜「三年游戏人」第一工夫浏览最新内容，获取一份收集多年的书籍包 以及 优质工作内推