[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.

给定二叉树的前序遍历以及中序遍历，要求构造该二叉树。

分析

这是典型的分而治之的题目。

二叉树的前序遍历的顺序是：节点-左子数-右子树，中序遍历的顺序是：左子数-节点-右子树，因此，通过前序遍历以及中序遍历，可以还原出原来的二叉树。

具体来说，前序遍历的第一个元素表示的就是根节点，那么找到根节点之后，在中序遍历中，位于根节点左边的元素便属于左子树，位于根节点右边的元素便属于右子树，于是将问题转化成了相同的两个子问题。问题的平凡形式是某边的子树为空的情况。

代码

/**
 * 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<int>& preorder, vector<int>& inorder) {
        return build(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
    }

    TreeNode* build(vector<int>& pre, int lo1, int hi1, vector<int>& in, int lo2, int hi2) {
        if (lo1 > hi1 || lo2 > hi2) return NULL;
        TreeNode* node = new TreeNode(pre[lo1]);

        int i = 0;
        for (i = lo2; i <= hi2; i++)
            if (in[i] == pre[lo1]) break;

        node->left = build(pre, lo1 + 1, i + lo1 - lo2, in, lo2, i - 1);
        node->right = build(pre, i + lo1 - lo2 + 1, hi1, in, i + 1, hi2);

        return node;
    }
};