首页 > 其他 > 详细

Construct Binary Tree from Preorder and Inorder Traversal

时间:2015-12-01 01:42:09      阅读:236      评论:0      收藏:0      [点我收藏+]

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

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

/**
 * 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, int preorderStart,int preorderEnd,vector<int>& inorder,int inorderStart,int inorderEnd) {
        if(preorderEnd-preorderStart==1){
            return new TreeNode(preorder[preorderStart]);
        }
        if(preorderEnd-preorderStart==0){
            return 0;
        }
        int val = preorder[preorderStart];
        int i=inorderStart;
        for(;i<inorderEnd;i++){
            if(inorder[i] == val){
                break;
            }
        }
        TreeNode *root = new TreeNode(val);
        int leftLen = i-inorderStart;
        int rightLen = inorderEnd - i - 1;
        root->left = buildTree(preorder,preorderStart+1,preorderStart+1+leftLen,inorder,inorderStart,inorderStart+leftLen);
        root->right = buildTree(preorder,preorderStart+1+leftLen,preorderEnd,inorder,inorderStart+leftLen+1,inorderEnd);
        return root;
        
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int preorederSize = preorder.size();
        int inorederSize = inorder.size();
        if(preorederSize != inorederSize){
            return NULL;
        }
        return buildTree(preorder,0,preorder.size(),inorder,0,inorder.size());
    }
};

 

Construct Binary Tree from Preorder and Inorder Traversal

原文:http://www.cnblogs.com/zengzy/p/5008922.html

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