首页 > 其他 > 详细

LeetCode OJ:Construct Binary Tree from Preorder and Inorder Traversal

时间:2014-01-23 04:41:51      阅读:336      评论:0      收藏:0      [点我收藏+]

Construct Binary Tree from Preorder and Inorder Traversal

 

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

Note:
You may assume that duplicates do not exist in the tree.


算法思想:

1、先序序列的第一个元素是该树的根;

2、中序序列该元素将该树分为左右两个子树,该元素位置设为i

3、中序序列的0到i-1元素为左子树,对应于先序序列的1到i元素,中序右子树i+1到in_end对应先序的i+1到pre_end元素

所以对于左子树和右子树也满足以上性质,递归即可

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void build(vector<int> &preorder, vector<int> &inorder, TreeNode * &root, int pre_start, int pre_end,int in_start, int in_end){
        if(in_start>in_end)return;
        root = new TreeNode(preorder[pre_start]);
        int i;
        for(i = in_start; i<=in_end; i++)
            if(inorder[i]==preorder[pre_start])
                break;
        int pre_mid = pre_start+i-in_start;
        build(preorder, inorder, root->left, pre_start+1, pre_mid, in_start, i-1);
        build(preorder, inorder, root->right, pre_mid+1, pre_end, i+1, in_end);
    }
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        if(!inorder.size())return NULL;
        TreeNode *root = NULL;
        build(preorder, inorder, root, 0, preorder.size()-1, 0, inorder.size()-1);
        return root;
    }
};


LeetCode OJ:Construct Binary Tree from Preorder and Inorder Traversal

原文:http://blog.csdn.net/starcuan/article/details/18667083

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!