面试题6:重建二叉树

关于重建二叉树这块,看《剑指offer》三四遍还是不能理解,然后今天中午又再次去领悟这个重建的思想,发现是自己想复杂了。遂决定写博客记录下自己的心得。

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

在这里它给的是两个序列,那么首先我们需要明确,前序遍历是根-左-右,中序遍历是左-根-右,那么首先可以根据前序遍历的序列确定根结点,随着根结点的确定,可以顺利找出左右子树。接下来,就不要考虑太多了,现在又相当于给你了一个新的前序遍历序列和中序遍历序列,我们再次重复上面步骤即可。

这里写图片描述

看这张图是不是觉得很清楚了,所以基于这个问题,我们只需要把这个问题一步步细化,将这个序列逐步分解,直到最后只剩一个元素就很清楚这是哪个子树的哪个结点。代码如下。

/**
 * 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 insize=vin.size();
        if(insize==0)
            return NULL;
        vector<int> preStart,preEnd;
        vector<int> inStart,inEnd;
        TreeNode* root=new TreeNode(pre[0]);  //前序遍历第一个肯定是根结点

        int rootpos=0;
        while(vin[rootpos]!=pre[0])   //在中序遍历中找到根的位置
            {
            rootpos++;
        }

        //rootpos前都是左子树,后面的都是右子树
        int i=0;
        for(;i<rootpos;i++)
            {
            preStart.push_back(pre[i+1]);
            inStart.push_back(vin[i]);
        }

        for(i=rootpos+1;i<insize;i++)
            {
            preEnd.push_back(pre[i]);
            inEnd.push_back(vin[i]);
        }

        root->left=reConstructBinaryTree(preStart,inStart);
        root->right=reConstructBinaryTree(preEnd,inEnd);

        return root;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树的中序线索化是将二叉树中的每个节点都指向其中序遍历的前驱和后继节点。这个过程需要对二叉树进行中序遍历,并将每个节点的左右指针修改为指向其前驱和后继节点。下面是中序线索化的实现代码: ```python class Node: def __init__(self, val, left=None, right=None): self.val = val self.left = left self.right = right self.left_thread = None self.right_thread = None def inorder_threading(root): if root is None: return stack = [] cur = root pre = None while cur or stack: while cur: stack.append(cur) cur = cur.left cur = stack.pop() if cur.left is None: cur.left = pre cur.left_thread = True if pre and pre.right is None: pre.right = cur pre.right_thread = True pre = cur cur = cur.right if not cur.right_thread else None if __name__ == '__main__': root = Node(1, Node(2), Node(3)) inorder_threading(root) print(root.left.val) # 2 print(root.left.right) # 1 print(root.left.right_thread) # True print(root.right.left) # 1 print(root.right.left_thread) # True print(root.right.val) # 3 ``` 上面的代码中,我们定义了一个 `Node` 类表示二叉树的节点。`left_thread` 和 `right_thread` 分别表示当前节点的左右指针是否是线索。初始时,二叉树的所有节点的 `left_thread` 和 `right_thread` 都是 `None`。 `inorder_threading` 函数实现了中序线索化的过程。我们使用一个栈来辅助进行中序遍历。对于每个节点,如果其左子节点为空,则将其左指针指向其中序遍历的前驱节点,并将 `left_thread` 标记为 `True`。如果其前驱节点的右指针为空,则将其右指针指向当前节点,并将 `right_thread` 标记为 `True`。 最后,我们可以验证中序线索化的结果是否正确。在上面的代码中,我们打印了一些节点的值和线索情况。可以看到,对于一个节点,其左指针指向的是其中序遍历的前驱节点,而右指针则指向其中序遍历的后继节点。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值