[剑指offer]重建二叉树

题目描述

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

分析：前序遍历列表中的第一个数就是root，在中序遍历列表中找到root元素，左边的部分为左子树，右边的部分为右子树，再分别对两部分递归构建，要注意子列表的起始下标位置

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:
    struct TreeNode* helper(vector<int> pre,vector<int> in,int ps,int pe,int is,int ie){
        if(ps>pe||is>ie) return nullptr;
        TreeNode* root=new TreeNode(pre[ps]);

		root->left=nullptr;
        root->right=nullptr;
        
            
        int i=is;
        for(;i<=ie;i++){
            if(in[i]==pre[ps]){
                break;
            }
        }
        root->left=helper(pre,in,ps+1,ps+i-is,is,i-1);
        root->right=helper(pre,in,ps+i-is+1,pe,i+1,ie);
        return root;
    }
    struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {
		int m=pre.size();
        int n=in.size();
        return helper(pre,in,0,m-1,0,n-1);
    }
};

python:

# -*- 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):
        if not pre:
            return None
        root=TreeNode(pre[0])
        pos=tin.index(pre[0])
        root.left=self.reConstructBinaryTree(pre[1:pos+1],tin[0:pos])
        root.right=self.reConstructBinaryTree(pre[pos+1:],tin[pos+1:])
        return root