剑指offer 重建二叉树 Python and C++

2020-01-08  本文已影响0人  GhostintheCode

题目描述

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

思路

1、首先需要理解前序中序后序的规则,前序是先根后左右,中序是先左再根再右,后序是先左右后跟。记忆的方法就是跟是在前还是在中还是在后。
2、又到了比较头疼的递归环节,我们只需要着眼于当前层。中需要把给的树,分成左右子树,然后把最后连接好的左右子树的根返回回去。
3、那么如何得到左右子树呢?这里有一点点小技巧,前序遍历的第一个节点,就是根节点,前序的左子树是从1到n的位置,n+1取不到,中序遍历根节点的位置左边就是所要构建的左子树,n的位置为根节点,n+1到最后为右子树。

做题可能出现的问题

python可能出现的问题

1、因为是树的问题,肯定是需要建立TreeNode的,
2、给的pre和tin都是数组,可以利用index找到根节点的位置,前提是里面不含有重复的数字。
3、递归的时候,要记得使用self因为这是在类里面。

C++可能出现的问题

1、TreeNode* node = new TreeNode(value);
2、int p需要写在外面面,而不能写在for的小括号里面,那样就变成局部变量了
3、c++没有python那么多语法糖,不能用slice的方式来,那么就用嵌套if语句来实现

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):
        # write code here
        if not pre:
            return None
        if not tin:
            return None
        root = TreeNode(pre[0])
        n = tin.index(root.val)
        root.left = self.reConstructBinaryTree(pre[1:n+1], tin[:n])
        root.right = self.reConstructBinaryTree(pre[n+1:], tin[n+1:])
        return 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:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int in_size = vin.size();
        if (in_size == 0)
        {
            return NULL;
        }
        vector<int> pre_left, pre_right, vin_left, vin_right;
        int value = pre[0];
        TreeNode* node = new TreeNode(value);
        int p;
        for(p=0; p<in_size;p++)
        {
            if (vin[p]==value)
                break;
        }
        for (int i = 0; i<in_size; i++)
        {
            if(i<p)
            {
                vin_left.push_back(vin[i]);
                pre_left.push_back(pre[i+1]);
            }
            else if(i>p)
            {
                vin_right.push_back(vin[i]);
                pre_right.push_back(pre[i]);
            }
        }
        node->left = reConstructBinaryTree(pre_left, vin_left);
        node->right = reConstructBinaryTree(pre_right, vin_right);
        return node;
    }
};
上一篇下一篇

猜你喜欢

热点阅读