LeetCode 105. Construct Binary Tree from Preorder and Inorder Traversal
本博客转载自:http://www.cnblogs.com/grandyang/p/4296500.html
Solution1:
递归,查好数是关键
/**
* 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 &preorder, vector &inorder) {
return my_buildTree(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
}
TreeNode *my_buildTree(vector &preorder, int pLeft, int pRight, vector &inorder, int iLeft, int iRight) {
if (pLeft > pRight || iLeft > iRight) return NULL;
int i = 0;
for (i = iLeft; i <= iRight; ++i) {
if (preorder[pLeft] == inorder[i]) break;
}
TreeNode *cur = new TreeNode(preorder[pLeft]);
cur->left = my_buildTree(preorder, pLeft + 1, pLeft + i - iLeft, inorder, iLeft, i - 1);//i - iLeft的值是当前inorder中属于右子树元素的个数
cur->right = my_buildTree(preorder, pLeft + i - iLeft + 1, pRight, inorder, i + 1, iRight);
return cur;
}
};