剑指 Offer 07-重建二叉树

题目

事先给定的代码

public class TreeNode
{
	public int val;
	public TreeNode left;
	public TreeNode right;

	public TreeNode(int x)
	{
		val = x;
	}
}

思路

解法一

先来看给出的示例。前序遍历是根左右，中序遍历是左根右。

根据前序数组，我们可以知道根节点是3。根据中序数组，我们可以推出3的左子树包含9，右子树包含15,20,7。

也就是说以3为根节点的树，左子树有1个元素，右子树有3个元素。那么再代回前序数组，就可以得知根节点往后1个元素，属于左子树，再往后3个元素属于右子树。

接下来我们就可以把[9]和[20,15,7]当成单独的前序数组，然后循环上述过程，就可以得出每棵子树的根节点对应的元素，进而也就可以构建出原本的二叉树了。
代码如下：

private Dictionary<int, int> _dic = new();
private int[] _preorder;
private int[] _inorder;
public TreeNode BuildTree(int[] preorder, int[] inorder)
{
	if (preorder == null  || inorder == null)
		return null;
	int preLen = preorder.Length;
	int inLen = inorder.Length;
	if (preLen == 0 || inLen == 0)
		return null;
	_preorder = preorder;
	_inorder = inorder;
	// 将中序数组的 元素-下标 存入字典
	for (int i = 0; i < inorder.Length; i++)
	{
		_dic[inorder[i]] = i;
	}
	return GetHead(0,inLen-1,0,preLen-1);
}

private TreeNode GetHead(int inLeft,int inRight,int preLeft,int preRight)
{
	// 左指针超过右指针,跳出递归
	if (preLeft > preRight)
		return null;
	// 前序数组第一个元素是整棵树的根节点
	TreeNode node = new TreeNode(_preorder[preLeft]);
	// 根节点位置
	int root = _dic[_preorder[preLeft]];
	// 左树长度
	int leftLen = root - inLeft;
	node.left = GetHead(inLeft, root - 1, preLeft + 1, preLeft + leftLen);
	node.right = GetHead(root+1,inRight,preLeft+leftLen+1,preRight);
	return node;
}