剑指Offer07.重建二叉树_解题思路&代码实现

一、题目描述

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

二、示例输入输出

示例1：

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

三、题目解析 & 解题思路

思路可知输入为前序遍历、中序遍历
前序遍历：前序遍历顺序为 根左右 即 先访问根，再遍历左子树，再遍历右子树
中序遍历：中序遍历顺序为 左根右 即 先访问左子树，再访问根节点，最后访问 右子树
前序遍历顺序特点：第一个 为根节点 例如 示例 1 为: 3 9 20 15 7 可以得知 节点 3 为整棵二叉树的根节点，再结合中序遍历 的顺序 ：9 3 15 20 7 可以得知 节点 3 左侧的 为整棵树的 左子树，节点 3 右边的为整棵树的 右子树
首先可以确定的就是根节点 为 前序遍历的第一个
如下图示例：

根据前序遍历特点：根左右 那么 前序遍历的结果中 左子树为 245 恰好 可以根据 中序遍历 中的根节点 1 划分 左子树 为 245 和 367 接下来 开始 根据前序 的 245 和 中序 的 425 来 判断 节点 2、4、5 这三个节点 那个是左孩子那个是右孩子即可

3.1

首先将 前序遍历中的第一个节点 作为根节点 插入到 二叉树中，若前序遍历为空 那么整个树也为空

3.2

另外 我们需要一个 栈进行辅助，保存前序遍历的节点(根节点或左孩子节点 因为 前序 根左右的 遍历顺序 会一直，根 -> 左(父) -> 左(父) 直到遍历到最后一个 左孩子 叶子节点，才会 回溯 回去 遍历 上一层的 右孩子节点)，且 右孩子节点肯定是 栈中 某一个节点的 右孩子节点
并从下标 index = 0 判断中序遍历 下标为 index 的元素 是否等于当前栈顶元素 如果等于栈顶元素，说明该子树 左孩子或父节点 已经遍历完毕

例如 假如 前序遍历和中序遍历的第一个节点都是 1 那么说明，这棵树没有左子树：

按照这样的思路 我们先 入栈 节点 124
stack [1, 2 ,4]
并且将这些节点 作为 左孩子 逐步插入，当遍历 到 前序 1245367 中的 5 时 我们发现 现在的栈顶 元素为 4 且 中序遍历顺序的第一个 元素也为 4 ，且中序遍历为 左根右 ，因此 节点 4 没有左孩子，那么在前序遍历 1245367 节点 4 的 下一个元素 5 ,肯定是 124 某一个节点 的 右孩子节点
index 依次向后移动，并和栈顶元素比较，因为栈里都是 根 左 根 左 ，栈顶到 栈底的顺序 为 左 根 左 根 ，而中序为 左 根 右 ，当 中序 下标为 index 且 与 栈顶元素相等 时，index 继续向后移动，栈 弹出 栈顶元素，当栈顶元素不等于 index 元素 时 说明遇到了 右孩子，且该节点是 上一次弹出的 节点的右孩子
即 5 != 1 则 5 为上一个弹出 的 节点 2 的 右孩子，这时 我们将 5 作为 右孩子 插入

总体思路：在前序中 构建节点的左孩子，中序中 确定 右孩子节点

四、代码实现

/**
 * 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) {
        if(preorder.size() == 0 || inorder.size() == 0)
        {
            return nullptr;
        }
        stack<TreeNode*> stkTree;
        //前序遍历结果的第一个作为 根节点插入
        TreeNode* Root = new TreeNode(preorder[0]);
        stkTree.push(Root);
        int inorderIndex = 0;
        for(int i = 1; i < preorder.size(); i++)
        {
            int iTemp = preorder[i];
            TreeNode* nodeTemp = stkTree.top();
            //从前序中构建出来左孩子节点 
            if(nodeTemp->val != inorder[inorderIndex])
            {
                TreeNode* node = new TreeNode(iTemp);
                nodeTemp->left = node;

                stkTree.push(nodeTemp->left);
            }
            else
            {
            	//从后序节点中找到 右孩子节点
                while(!stkTree.empty() && (stkTree.top()->val == inorder[inorderIndex]))
                {
                	//保存节点，最后一个弹出的为右孩子的父节点
                    nodeTemp = stkTree.top();
                    stkTree.pop();
                    ++inorderIndex;
                }
                nodeTemp->right = new TreeNode(iTemp);
                stkTree.push(nodeTemp->right);
            }
        }
        return Root;

    }
};