题目:
根据二叉树的前序遍历和中序遍历的结果,重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。分析
- 前序遍历顺序:根节点->左子节点->右子节点
- 中序遍历顺序:左子节点->根节点->右子节点
前序中的第一个元素能把中序分成两部分,然后对左右两部分做同样操作,必然使用递归方法,递归出口,在最后只有一个元素,递归结束
思路
源码
package org.example.tree;
import org.example.model.TreeNode;
import java.util.Arrays;
/**
* 重建二叉树
* @author Abner G
* @date 2022/5/14 15:55
*/
public class practiceTree01 {
/**
* 题目
* 根据二叉树的前序遍历和中序遍历的结果,重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
*
* https://github.com/CyC2018/CS-Notes/blob/master/notes/7.%20%E9%87%8D%E5%BB%BA%E4%BA%8C%E5%8F%89%E6%A0%91.md
*
* 分析
*
* 前序遍历顺序:根节点->左子节点->右子节点
* 中序遍历顺序:左子节点->根节点->右子节点
*
* 前序中的第一个元素能把中序分成两部分
* 然后对左右两部分做同样操作,必然使用递归方法
* 递归出口,在最后只有一个元素,递归结束
*
*/
/**
*
* @param pre 前序排列数组
* @param ino 中序排列数组
* @return
*/
public TreeNode reConstructBinaryTree(int[] pre,int[] ino){
if (pre.length==0){
return null;
}
//前序第一个元素
int rootVal = pre[0];
//将中序分成左右子树两部分
TreeNode treeNode = new TreeNode(rootVal);
int rootIndex=0;
for (int i = 0; i < ino.length; i++) {
if (rootVal==ino[i]){
rootIndex=i;
break;
}
}
//左子树
//前序 copyOfRange 范围0<=X<10
Arrays.copyOfRange(pre, 1, rootIndex + 1);
//中序
Arrays.copyOfRange(ino, 0, rootIndex );
treeNode.setLeftNode(reConstructBinaryTree(Arrays.copyOfRange(pre, 1, rootIndex + 1),Arrays.copyOfRange(ino, 0, rootIndex )));
//同理右子树
Arrays.copyOfRange(pre, rootIndex + 1, pre.length);
Arrays.copyOfRange(ino, rootIndex+1, ino.length );
treeNode.setRightNode(reConstructBinaryTree(Arrays.copyOfRange(pre, rootIndex + 1, pre.length),Arrays.copyOfRange(ino, rootIndex+1, ino.length )));
return treeNode;
}
}