LeetCode106. 从中序与后序遍历序列构造二叉树(Java）

题目

根据一棵树的中序遍历与后序遍历构造二叉树。你可以假设树中没有重复的元素。

例如，给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

HashMap<Integer, Integer> map = new HashMap<>();// 用于存储中序遍历的元素
	int index;// 从后序遍历最后一个元素(根）开始，往前移动

	public TreeNode buildTree(int[] inorder, int[] postorder) {
		for (int i = 0; i < inorder.length; i++) {
			map.put(inorder[i], i);// 找根节点的下标方便
		}
		index = postorder.length - 1;
		return dfs(0, inorder.length - 1, inorder, postorder);
	}

	// 找到根节点的位置，划分出左右子树，再从左右子树中找根节点，如此循环
	public TreeNode dfs(int left, int right, int[] inorder, int[] postorder) {
		if (index < 0 || left < 0 || left > right) {// 如果left>right说明左右子树已形成
			return null;
		}
		int val = postorder[index--];// 根节点的值，index要不断往前移动，以指向不同的根节点
		TreeNode root = new TreeNode(val);// 在这里生成一个root节点，递归下去可以创建节点之间的联系
		int tempIndex = map.get(val);// 在中序遍历中根节点的位置
		root.right = dfs(tempIndex + 1, right, inorder, postorder);// 右子树
		root.left = dfs(left, tempIndex - 1, inorder, postorder);// 左子树
		return root;
	}

细节：

1. 根的创建

TreeNode root = new TreeNode(val);// 在这里生成一个root节点，递归下去可以创建节点之间的联系

2. 是先从postorder里面取出元素，再去map里找下标：
   int val = postorder[index–];
   int tempIndex = map.get(val);

第二天review

public class 从中序与后序遍历序列构造二叉树 {
	HashMap<Integer, Integer> map = new HashMap<>();
	int index;// 设置一个变量从最后一个开始往前指,以找到以该节点为根的子树

	public TreeNode buildTree(int[] inorder, int[] postorder) {
		// 1.HashMap存中序遍历的元素与下标的关系
		// 2.通过后序遍历找根节点的的值，再在中序遍历中找到此根节点的下标，从而划分左右子树
		// 后序遍历左右根，最后一个是根节点，所以设置一个变量从最后一个开始往前指
		// 3.递归创建数，先创建树根,给树根赋值，然后再分别讨论左右子树
		// 4.如何创建左右子树：通过根节点的下标划分左右子树的范围，递归下去，范围逐渐缩小
		for (int i = 0; i < inorder.length; i++)
			map.put(inorder[i], i);
		index = postorder.length - 1;
		return helper(inorder, postorder, 0, inorder.length - 1);

	}

	public TreeNode helper(int[] inorder, int[] postorder, int left, int right) {
		if (left > right) {
			return null;
		}
		int val = postorder[index--];
		TreeNode root = new TreeNode(val);
		root.right = helper(inorder, postorder, map.get(val) + 1, right);// 后序变量左右根，倒着来，先根，然后右子树，最后左子树
		root.left = helper(inorder, postorder, left, map.get(val) - 1);
		return root;
	}
}

反思

我太菜了
end.