4.重建二叉树

题目链接

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

算法背景：

前序遍历递归算法：NLR 先访问根结点N，再遍历左子树L，再遍历右子树R。

中序遍历递归算法：LNR 先访问左子树L，再访问根结点，再遍历右子树R。

根据前序遍历和中序遍历可以确定二叉树：

1. 前序序列第一个结点确定二叉树的根结点
2. 根据根结点在中序序列中的位置，前部分为根结点的左子树，后部分分为根结点的右子树。
3. 再对分割出来的左子树和右子树进行递归分析。

案例：

前序遍历序列为ABCDEFGHI，中序遍历序列为BCAEDGHFI。

pre序列首元素必为二叉树的根结点，在vin序列中找到相应位置gen确定根结点，由于vin是LNR的特征，gen前的序列为左子树中序序列，gen后的序列为右子树中序序列。同样在pre序列中，gen+1前的序列为左子树的前序序列，gen+1后的序列为右子树的前序序列，然后利用子树的前序序列和中序序列递归求出二叉树。

算法：

1. 由先序序列第一个**pre[0]在中序序列中找到根节点位置gen**

2. 以gen为中心遍历

   • 0~gen

     左子树

     • 子中序序列：0~gen-1，放入**vin_left[]**
     • 子先序序列：1~gen放入pre_left[]，**+1**可以看图，因为头部有根节点

   • gen+1~vinlen

     为右子树

     • 子中序序列：gen+1 ~ vinlen-1放入vin_right[]
     • 子先序序列：gen+1 ~ vinlen-1放入pre_right[]

3. 由先序序列**pre[0]**创建根节点

4. 连接左子树，按照左子树子序列递归（pre_left[]和vin_left[]）

5. 连接右子树，按照右子树子序列递归（pre_right[]和vin_right[]）

6. 返回根节点

时间复杂度O(n)，空间复杂度O(n)。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int vinlen = vin.size();
        if (vinlen == 0)
        {
            return NULL;
        }
        vector<int> pre_left, pre_right, vin_left, vin_right;
        TreeNode* root = new TreeNode(pre[0]);
        int index = 0;
        for (int i = 0; i < vinlen; ++i)
        {
            if (vin[i] == pre[0])
            {
                index = i;
                break;
            }
        }
        for (int i = 0; i < index; ++i)
        {
            pre_left.push_back(pre[i + 1]);
            vin_left.push_back(vin[i]);
        }
        for (int i = index + 1; i < vinlen; ++i)
        {
            vin_right.push_back(vin[i]);
            pre_right.push_back(pre[i]);
        }
        root->left = reConstructBinaryTree(pre_left, vin_left);
        root->right = reConstructBinaryTree(pre_right, vin_right);
        return root;
    }
};