您的位置:首页 > 其它

LeetCode | Construct Binary Tree from Preorder and Inorder Traversal

2013-12-17 20:32 381 查看


题目:

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

Note:

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


思路:

/article/1382549.html类似,但是过程相反。


代码:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> preorder;
    vector<int> inorder;
    int cur;
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        if(preorder.size() == 0)
            return NULL;
        else
        {
            cur = 0;
            this->preorder = preorder;
            this->inorder = inorder;
            return buildSubTree(0, inorder.size()-1);
        }
    }
    
    TreeNode* buildSubTree(int from, int to)
    {
        TreeNode * root = new TreeNode(preorder[cur]);
        int index = findValInInOrder(preorder[cur]);
        cur++;
        if(index <= from)
        {
            root->left = NULL;
        }
        else
        {
            root->left = buildSubTree(from, index-1);
        }
        if(index >= to)
        {
            root->right = NULL;
        }
        else
        {
            root->right = buildSubTree(index+1,to);
        }
        return root;
    }
    
    int findValInInOrder(int val)
    {
        for(int i=0;i<inorder.size();i++)
        {
            if(val == inorder[i])
            {
                return i;
            }
        }
        return -1;
    }
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐