LeetCode: Construct Binary Tree from Preorder and Inorder Traversal [105]

【题目】

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 binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    
    int getPosOfRoot(vector<int>&inorder, int istart, int iend, int key){
        for(int i=istart; i<=iend; i++){
            if(inorder[i]==key)return i;
        }
    }
    
    TreeNode *build(vector<int>&preorder, int pstart, int pend, vector<int>&inorder, int istart, int iend){
        if(pstart>pend)return NULL;
        // 创建根节点
        TreeNode*root = (TreeNode*)malloc(sizeof(TreeNode));
        root->val=preorder[pstart];
        
        //确定根节点在中序遍历中的位置
        int rootPos = getPosOfRoot(inorder, istart, iend, preorder[pstart]);
        //左子树中序遍历序列的长度
        int lenLeft = rootPos-istart;
        //右子树中序遍历序列的长度
        int lenRight = iend-rootPos;
        //创建左子树
        TreeNode* left=build(preorder, pstart+1, pstart+lenLeft, inorder, istart, rootPos-1);
        //创建右子树
        TreeNode* right=build(preorder, pstart+lenLeft+1, pend, inorder, rootPos+1, iend);
        
        //构造二叉树
        root->left=left;
        root->right=right;
        return root;
    }

    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        int size=preorder.size();
        if(size==0)return NULL;
        
        return build(preorder, 0, size-1, inorder, 0, size-1);
        
    }
};