[面试真题] LeetCode：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时，返回NULL。

　　递归步骤：首先确定preorder的第一个元素一定是该树的root，再在inorder中找到该元素，标记为index，index左部为左子树的中序，右部为右子树的中序；随后通过左右子树中序的长度（可能为0），在preorder中确定左右子树的先序。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    TreeNode *buildTree(vector<int> &preorder,int ph, int pe, vector<int> &inorder, int ih, int ie){
        TreeNode *root;
        if(pe - ph < 0){
            return NULL;
        }
        root = new TreeNode(preorder[ph]);
        int index = ih-1;
        while(++index <= ie && inorder[index] != preorder[ph]){
        }
        int leftPreLen = index - ih;
        root->left = buildTree(preorder, ph+1, ph+leftPreLen, inorder, ih, index-1);
        root->right = buildTree(preorder, ph+leftPreLen+1, pe, inorder, index+1, ie);
        return root;
    }
    
public:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        return buildTree(preorder, 0, preorder.size()-1, inorder, 0, inorder.size()-1);
        
    }
};

 

Run Status: Accepted!
Program Runtime: 164 milli secs