105. 从前序与中序遍历序列构造二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7
   
   限制:

0 <= 节点个数 <= 5000
为什么需要已知前序遍历和中序遍历??

因为只知道前序遍历或只知道中序遍历,是无法确定左子树、根、右子树的。因为前序遍历顺序为根左右,中序遍历为左根右。

思路

前序遍历数组的从左往右是一系列顺序排列的节点
找出根节点,得到根节点的val
从前序遍历中取出一个数,再去中序遍历中寻找相同的数,该数将中序遍历数组分割成了左右子树。
递归以上步骤,直到达到递归终止条件。

前序遍历的顺序

从根结点开始,进入左子树下层,再进入左子树,再进入左子树…直到左子树为空,再考虑空左子树的兄弟右子树。即使进入了兄弟右子树,也还是继续探索该右子树的左子树…

中序遍历的顺序

就是一条垂直线从左往右扫描二叉树(犹如一根垂直线扫过一个正三角形)。

/**
 * 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:
    map<int, int> indexTable;	// 方便直接获取inorder数组的索引
    int p = 0;	// 用于遍历整个preorder数组,使用全局变量,再递归时更新。
    // 本次递归中,以preorder[p]为根的二叉树的区间范围就是inorder[istart,iend),是一个左闭右开的区间,注意边界判断。
    TreeNode* dfs(vector<int>& preorder, vector<int>& inorder, int istart, int iend){
        // printf("p=%d, istart=%d, iend=%d\n", p, istart, iend);
        if(p >= preorder.size() || (istart >= iend)) return NULL;
        // 创建根结点,以preorder[p]为根
        TreeNode* curNode = new TreeNode(preorder[p]);
        
        // 处理当前根结点的左子树,inorder[start, root)都属于左子树,inorder[root + 1, end)都属于右子树
        // 获取根结点在inorder中的位置,inorder[root]就是根
        int root = indexTable[preorder[p]];
        p += 1;
        curNode->left = dfs(preorder, inorder, istart, root);
        
        // 再处理右子树
        curNode->right = dfs(preorder, inorder, root + 1, iend);
        
        return curNode;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {

        for (int i = 0; i < inorder.size(); ++i)
            indexTable[inorder[i]] = i;
        // inorder的区间是左闭右开
        return dfs(preorder, inorder, 0, inorder.size());
        
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值