【剑指offer】04——重建二叉树

题目描述

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

一、二叉树的遍历

  二叉树遍历分为三种：前序、中序和后序，为什么这么命名呢？其实是根据节点顺序命名的。

如图为满节点，1为根节点、2为做节点、3为右节点，主要是看根结点1的位置，在前面就是前序遍历、在中间就是中序遍历、在后面就是后续遍历。

下面我们通过例子，来看看：

1、前序遍历

• 在遍历节点的左节点的时候，可能左边节点还有左节点和右节点，要一直遍历到叶子节点，才能遍历其右节点
• 例如在遍历到节点2时，它的左节点是3，但是节点3不是叶子节点，此时就不能去遍历节点2的右节点4
• 虽然节点3没有左节点，但是有右节点5，它应该在节点2的右节点4的前面
• 也就是我们在遍历的时候，要将左边节点遍历完才能遍历右边节点

2、中序遍历

3、后序遍历

二、解题思路

  前序遍历的第一个值为根结点的值，使用这个值将中序遍历结果分成两部分，左部分为数的左子树中序遍历结果，右部分为树的右子树中序遍历的结果。

很明显，上面出现的类似的操作，只是输入的前序和中序结果不一样，我们可以使用递归的方式

三、Java代码

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode resultTree = myConstructBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
        //pre:前序遍历结果，0开始位置，pre.length-1结束位置
        //in:中序遍历结果，0开始位置，in.length-1结束位置
        return resultTree;
    }
    
    private TreeNode myConstructBinaryTree(int[] pre,int startPre,int endPre,int[] in, int startIn,int endIn){
    	//如果开始位置大小结束位置，返回null
        if((startPre > endPre) || (startIn > endIn)){
            return null;
        }
        //新建一个树，根结点为pre[startPre]，即前序遍历中的第一个元素
        TreeNode myTree = new TreeNode(pre[startPre]);
        for(int i=startIn;i<=endIn;i++){  //遍历中序遍历结果
            if(in[i] == pre[startPre]){ //判断中序遍历中与前序遍历第一个元素相同的，然后分为两个部分，进行递归
                myTree.left = myConstructBinaryTree(pre, startPre+1, startPre+i-startIn, in, startIn, i-1);
                myTree.right = myConstructBinaryTree(pre, startPre+i+1-startIn, endPre, in, i+1, endIn);
            }
        }
        return myTree;
    }
}

四、C++代码

/**
 * 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) {
        if(pre.size() == 0){  //如果为空，返回NULL
            return NULL;
        }
        //依次是前序遍历左子树，前序编写右子树，中序遍历左子树，中序遍历右子树
        vector<int> left_pre,right_pre,left_vin,right_vin;
        //前序遍历第一个节点一定为根结点
        TreeNode* head = new TreeNode(pre[0]);
        //找到中序遍历的根结点
        int root = 0;
        //遍历找到中序遍历根结点的索引
        for(int i=0; i<pre.size();i++){
            if(pre[0] == vin[i]){
                root = i;
                break;
            }
        }
        
        //利用中序遍历的根结点，将二叉树分为两部分，左子树和右子树
        for(int i=0; i < root; i++){
            left_vin.push_back(vin[i]);
            left_pre.push_back(pre[i+1]); //前序遍历的第一个为根结点
        }
        
        for(int i=root+1;i<pre.size();i++){
            right_vin.push_back(vin[i]);
            right_pre.push_back(pre[i]);
        }
        
        //递归
        head->left = reConstructBinaryTree(left_pre,left_vin);
        head->right = reConstructBinaryTree(right_pre,right_vin);
        
        return head;
    }
};

五、Python3代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        # 如果为空
        if not pre and not tin:
            return None
        if(set(pre) != set(tin)): # 前序和中序遍历结果中元素不一样
            return None
        root = TreeNode(pre[0]) # 根结点为前序遍历的第一个元素
        i = tin.index(pre[0])  # 根结点在中序遍历结果中的下标
        # 递归
        root.left = self.reConstructBinaryTree(pre[1:i+1],tin[:i])
        root.right = self.reConstructBinaryTree(pre[i+1:],tin[i+1:])
        return root