剑指 Offer 07. 重建二叉树

题目

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

给出:

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回:
3
/
9 20
/
15 7
其实就是【3,9,20,null,null,15,7】

递归法:
记录所有节点的索引,用字典或者unordered_map存储
找到根节点-》分开左右子树-》递归遍历,返回节点
递归出口,左边界大于右边界
python版本

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
	def buildTree(self,preorder,inorder):
		def recur(root,left,right):
			if left>right:
				node=TreeNode(preorder[root]) #选根节点
				i=dic[preorder[root]]
				node.left=recur(root+1,left,i-1)
				node.right=recur(i-left+root+1,i+1,right)
				return node
		dic,preorder={},preorder
		#记录所有节点的索引位置
		for i in range(len(inorder)):
			dic[inorder[i]]=i
		return recur(0,0,len(inorder)-1)

C++版本

/**
 * 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){
		this->preorder=preorder;
		for(int i=0;i<inorder.size();i++){
			dic[inorder[i]]=i;
		}
		return recur(0,0,inorder.size()-1);
	}
private:
	vector<int> preorder;
	unordered_map<int,int> dic;
	TreeNode* recur(int boot,int left,int right){
		if(left>right) return nullptr;
		TreeNode* node=new TreeNode(preorder[root]);
		int i=dic[preorder[root]];
		node->left=recur(root+1,left,i-1);
		node->right=recur(i-left+root+1,i+1,right);
		return node;
	}
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值