关闭

Construct Binary Tree from Preorder and Inorder Traversal

标签: C++leetcode遍历二叉树递归
111人阅读 评论(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;
    }
};


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:123450次
    • 积分:4417
    • 等级:
    • 排名:第7023名
    • 原创:341篇
    • 转载:33篇
    • 译文:0篇
    • 评论:5条
    最新评论