美文网首页
剑指offer 重建二叉树 Python and C++

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

作者: GhostintheCode | 来源:发表于2020-01-08 18:23 被阅读0次

    题目描述

    输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
        }
    };
    

    相关文章

      网友评论

          本文标题:剑指offer 重建二叉树 Python and C++

          本文链接:https://www.haomeiwen.com/subject/fvjsactx.html