【剑指offer】重建二叉树

题目链接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/

题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

思路:一开始看到这个题,不知道是没信心的原因,还是以前做pat甲级题,遇到的都是根据前序中序转后序的题,很少有这种构建二叉树返回的题目,于是开始网上搜题解,但是搜了半天,题解都差强人意,写得很长并且难懂,于是凭借着乱搞的心态,居然自己写出来了,,,就是不断地去找根节点,然后构建出这个节点,再递归的去找这个节点的左子树和右子树。

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
vector<int> preorder, inorder;
public:
    TreeNode* build(int root, int s, int e){
        if(s > e) return NULL;
        int i = s;
        TreeNode* node = new TreeNode(preorder[root]);
        while(inorder[i] != preorder[root]) i++;
        node -> left = build(root + 1, s, i - 1);
        node -> right = build(root + i - s + 1, i + 1, e);
        return node;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        this -> preorder = preorder;
        this -> inorder = inorder;
        int n = preorder.size();
        TreeNode *tree = build(0, 0, n - 1);
        return tree;
    }
};

 

【剑指offer】重建二叉树【剑指offer】重建二叉树 Xinstein666 发布了12 篇原创文章 · 获赞 0 · 访问量 282 私信 关注
上一篇:ICPC2021上海区域赛 D.Walker(二分、分类讨论)


下一篇:二叉树的构建及遍历