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);            }};