面试题7：重建二叉树

面试题7：重建二叉树

题目描述

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

二叉树的定义

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };

思路

由前序遍历序列{1,2,4,7,3,5,6,8}，可以得到的信息：根节点为1

由1.中序遍历序列{4,7,2,1,5,3,8,6}；2.根节点为1；可以得到的信息：左子树为{4,7,2}，右子树为{5，3，8，6}

对左右子树再重复上述过程。

整个重建递归实现。

代码实现

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        //0.特殊情况
        if(pre.empty() || vin.empty())
            throw std::exception("Invalid input");
        
        return reConstructBinaryTree(pre,0,pre.size()-1,vin,0,vin.size()-1);        
    }
private:
    TreeNode* reConstructBinaryTree(vector<int> pre,int startPre,int endPre,
                                    vector<int> vin, int startIn, int endIn){
        //1.终止条件
     if(startPre>endPre||startIn>endIn)
            return NULL;
        
        //2.根节点
     TreeNode* root = new TreeNode(pre[startPre]);
        
        //3.划分左右子树 递归求解
     for(int i=startIn;i<=endIn;i++){
         if(vin[i]==pre[startPre]){//在中序遍历序列中找到根节点的值
     		 root->left =
          reConstructBinaryTree(pre,startPre+1,startPre+i,startIn,vin,startIn,i-1);
      		root->right =
          reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,vin,i+1,endIn);
            break;
         }
         
         //前序序列和中序序列不匹配
         if(i == endIn && vin[i]!=pre[startPre]){
             throw std::exception("Invalid input");
         }  
     }
     
      return root;
    }

};

考点

二叉树的前序遍历和中序遍历。

复杂问题（二叉树重建）划分为小问题（子树重建）递归求解。

Note

前序序列和中序序列不匹配的解决：中序遍历序列中查找根节点的值失败。

存在问题：查找成功但据此划分的左右子树和前序遍历不匹配。在后面的递归过程中会在上面的条件中抛出异常。

函数重载：形参类型或数量不同。与形参名字无关。返回类型不同不构成重载。