剑指 Offer(树之篇 1).[重建二叉树]
题目要求:
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
解题思路:
1.了解前序遍历和中序遍历的原理
- **前序遍历;**先访问根结点,然后再访问左子树,最后访问右子树
- **中序遍历;**先访问左子树,中间访问根节点,最后访问右子树
2.找到根结点在前序遍历数组preorder和中序遍历数组inorder中的位置
3.根据前序排序和中序排序的规律,分别使用递归总结规律得到树的左子结点和右子结点
对以上问题的详细解释都在代码中标注
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
//首先了解前序排序,先是树的根结点,再到左子结点后是右子结点,所以preorder[0]一定是根结点
int treeLength = preorder.length;
if(treeLength == 0){
return null;
}
//确定根结点root
int rootValue = preorder[0];
//明确中序排序的排序方法,先是左子结点,再是根结点,最后是右子结点
//找到根结点在中序排序中的位置,中序排序中根结点前面的都是左子结点,后面的是右子结点,直到下一个根结点
//通过树中根结点的值preorder[0]找到在inorder数组根结点rootIndex所在的位置
int rootIndex = 0;
for(int i = 0;i < treeLength;i++){
if(inorder[i] == rootValue){
rootIndex = i;
break;
}
}
//按照之前说的规律,对buildTree采取递归的方法将树中根结点的左子结点和右子结点全部找出来
//构建出新的二叉树TreeNode
TreeNode root = new TreeNode(rootValue);
//根据前序遍历的方法,在排序后的数组preorder,从索引1到rootIndex + 1都是子树x根结点的左子结点(左闭右开)
//根据中序遍历的方法,在排序后的数组inorder,从索引0到rootIndex都是子树x根结点的左子结点(左闭右开)
root.left = buildTree(Arrays.copyOfRange(preorder,1,rootIndex + 1),Arrays.copyOfRange(inorder,0,rootIndex));
//根据前序遍历的方法,在排序后的数组preorder,根结点在索引0位置,从索引rootIndex + 1到数组结尾preorder.length都是子树x根结点的右子结点(左闭右开)
//根据中序遍历的方法,在排序后的数组inorder,根结点在索引i位置,从索引rootIndex + 1到inorder.length都是子树x根结点的右子结点(左闭右开)
root.right = buildTree(Arrays.copyOfRange(preorder,rootIndex + 1,preorder.length),Arrays.copyOfRange(inorder,rootIndex + 1,inorder.length));
return root;
}
}