Leetcode 105

题意

根据二叉树的前序遍历结果和中序遍历结果重构二叉树。

思路

根据前序遍历和中序遍历的性质我们可以知道前序遍历的第一个节点一定是二叉树的根节点，中序遍历根节点之前的是其左子树，根节点之后的是其右子树。

于是我们在中序遍历的数组中找到当前位置pos，那么在中序遍历中，从0到pos - 1一定是二叉树的左子树，从pos + 1到n一定是二叉树的右子树，递归的去做就好。

代码

/** * 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* build(vector<int>& a, int l1, int r1, vector<int>& b, int l2, int r2) {        if (l1 == r1) {            TreeNode* node = new TreeNode(a[l1]);            return node;        } else if (l1 > r1) {            return NULL;        }        TreeNode* root = new TreeNode(a[l1]);        int off = 0;        for (int i = l2; i <= r2; i++) {            if (b[i] == a[l1]) {off = i - l2; break;}        }        root->left = build(a, l1 + 1, l1 + off, b, l2, l2 + off - 1);        root->right = build(a, l1 + off + 1, r1, b, l2 + off + 1, r2);        return root;    }    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {        return build(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);    }};