【LeetCode】Construct Binary Tree from Preorder and Inorder Traversal 根据先序序列和中序序列恢复二叉树

题目:Construct Binary Tree from Preorder and Inorder Traversal

<span style="font-size:18px;">/**LeetCode 
 * 题意:给定一个二叉树的先序遍历和中序遍历的序列,重建二叉树
 * 思路:先序遍历的第一个节点是父节点,中序遍历从第一个节点到父节点的都是父节点的左子树,父节点后面的都是父节点的右子树
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
package javaTrain;

public class Train20 {
	public TreeNode buildTree(int[] preorder, int[] inorder) {
		int n = preorder.length;
		if(n == 0) return null;
		int m = 0;
		int num = preorder[0];
		boolean tag = true;
		for(int i = 0;i < n;i++){ 
			if(inorder[i] == num)
				break;
			m++;
		}
		int[] leftPre = new int[m];
		int[] leftIn = new int[m];
		int[] rightPre = new int[m-n-1];
		int[] rightIn = new int[m-n-1];
		for(int j = 0;j < n;j++){
			if(j == num-1) tag = false;
			if(tag){
				leftIn[j] = inorder[j];
				leftPre[j] = preorder[j+1];
			}
			else{
				rightIn[j-num-1] = inorder[j];
				rightPre[j] = preorder[j+1];
			}
		}
		TreeNode root = new TreeNode(num);
		root.left = buildTree(leftPre,leftIn);
		root.right = buildTree(rightPre,rightIn);
		return root;
    }
}
</span>


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
我们可以通过递归的方式来实现给定先序序列和层序序列构造二叉树的过程。 具体实现步骤如下: 1.定义二叉树结构体 首先,我们需要定义二叉树结构体,包括节点值和左右子节点的指针。 ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right ``` 2.递归构造二叉树 接着,我们可以定义一个递归函数来构造二叉树。该函数的输入参数包括先序序列和层序序列,以及根节点在先序序列中的位置、子树在层序序列中的起始位置和结束位置。 ```python def buildTree(preorder, inorder): if not preorder: return None # 根节点为先序序列的第一个节点 root_val = preorder[0] root = TreeNode(root_val) # 划分左右子树 idx = inorder.index(root_val) left_inorder = inorder[:idx] right_inorder = inorder[idx+1:] left_preorder = [] right_preorder = [] # 根据左右子树在中序序列中的位置,划分先序序列 for val in preorder[1:]: if val in left_inorder: left_preorder.append(val) else: right_preorder.append(val) # 递归构造左右子树 left_start = 0 left_end = len(left_inorder) - 1 right_start = len(left_inorder) right_end = len(inorder) - 2 root.left = buildTree(left_preorder, left_inorder, left_start, left_end, levelorder) root.right = buildTree(right_preorder, right_inorder, right_start, right_end, levelorder) return root ``` 在递归函数中,我们首先根据先序序列的第一个节点来构造根节点。然后,根据根节点在中序序列中的位置,将中序序列划分为左右子树,并将先序序列也划分为左右子树。接着,我们递归构造左右子树,并将其连接到根节点上。 3.获取层序序列 最后,我们需要实现一个函数来获取给定二叉树的层序序列。该函数可以使用队列来实现广度优先遍历,将每层节点的值添加到结果列表中。 ```python def levelOrder(root): if not root: return [] queue = [root] res = [] while queue: level = [] size = len(queue) for i in range(size): node = queue.pop(0) level.append(node.val) if node.left: queue.append(node.left) if node.right: queue.append(node.right) res.append(level) return res ``` 完整代码实现如下: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def buildTree(preorder, inorder): if not preorder: return None # 根节点为先序序列的第一个节点 root_val = preorder[0] root = TreeNode(root_val) # 划分左右子树 idx = inorder.index(root_val) left_inorder = inorder[:idx] right_inorder = inorder[idx+1:] left_preorder = [] right_preorder = [] # 根据左右子树在中序序列中的位置,划分先序序列 for val in preorder[1:]: if val in left_inorder: left_preorder.append(val) else: right_preorder.append(val) # 递归构造左右子树 left_start = 0 left_end = len(left_inorder) - 1 right_start = len(left_inorder) right_end = len(inorder) - 2 root.left = buildTree(left_preorder, left_inorder, left_start, left_end, levelorder) root.right = buildTree(right_preorder, right_inorder, right_start, right_end, levelorder) return root def levelOrder(root): if not root: return [] queue = [root] res = [] while queue: level = [] size = len(queue) for i in range(size): node = queue.pop(0) level.append(node.val) if node.left: queue.append(node.left) if node.right: queue.append(node.right) res.append(level) return res ``` 参考链接: - [LeetCode 题解](https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/gong-shui-san-xie-yi-ti-shuang-jie-ti-rua-39f5/) - [算法基础课](https://www.acwing.com/video/2574/)

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值