Construct Binary Tree from Preorder and Inorder Traversal

最新推荐文章于 2024-09-19 08:57:00 发布

fullstack_lth

最新推荐文章于 2024-09-19 08:57:00 发布

阅读量261

点赞数

分类专栏： LeetCode 文章标签： C++ leetcode 遍历二叉树递归

本文链接：https://blog.csdn.net/u014744118/article/details/52368118

版权

LeetCode 专栏收录该内容

161 篇文章 0 订阅

订阅专栏

一、问题描述

Given preorder and inorder traversal of a tree, construct the binary tree.

二、思路

二叉树的题目用递归做是最便捷的，本题也不例外。

递归的退出条件是：

       if(pre_begin > preorder.size() - 1 || in_begin > in_end) return NULL;

最关键的是在先序遍历中找出当前节点的值，即root -> val。还有我们在循环中加入break，可以使我们的运行时间从20ms减少到16ms。

在执行递归的时:

左子树在先序遍历指向下一个节点，即左子树的起始位置，在中序遍历中的范围是（in_begin,index - 1）;

右子树在先序遍历指向是当前位置减去中序遍历开始的位置，即右子树的起始位置，中序遍历中的范围是（index + 1, in_end）;

三、代码

/**
 * 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 {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return creatTree(0, 0, inorder.size() - 1, preorder, inorder);
    }
    TreeNode* creatTree(int pre_begin, int in_begin, int in_end, vector<int>& preorder, vector<int>& inorder){
       if(pre_begin > preorder.size() - 1 || in_begin > in_end) return NULL;
       TreeNode *root = new TreeNode(preorder[pre_begin]);
       int index = 0; 
       for (int i = in_end; i >= in_begin; --i) {
            if (inorder[i] == root -> val) {
                index = i;
                break;
            }
       }
       root -> left = creatTree(pre_begin + 1, in_begin, index - 1, preorder, inorder);
       root -> right = creatTree(pre_begin + index - in_begin + 1, index + 1, in_end, preorder, inorder);
       return root;
    }
};