CoolSummer

数据可视化

Construct Binary Tree from Preorder and Inorder Traversal

一、问题描述

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;
    }
};


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u014744118/article/details/52368118
个人分类: LeetCode
上一篇Triangle
下一篇Construct Binary Tree from Inorder and Postorder Traversal
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭