思路
1.采用递归的方式,每次进入函数先将参数列表的pre[0]作为根节点。
2.如果pre数组里只有一个数就返回叶子节点
3.否则将中序数组基于根节点(in[count])分成左右两部分
- 左边的为一个数组(存放的都是根节点左边的节点),将先序数组分成对应的部分,进行递归
- 右边的为一个数组(存放的都是根节点右边的节点),将先序数组分成对应的部分,进行递归
4.返回第一个根节点
代码
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode tree = new TreeNode(pre[0]);
//数组里只有一个数则返回根节点
if (pre.length == 1) {
return tree;
}
int count = 0;
//计算根节点左边数组的长度
while (count < in.length && pre[0] != in[count]) {
count++;
}
//生成根节点左边的树
if (count != 0) {//判断左边的树存不存在
int[] p1 = new int[count];
int[] i1 = new int[count];
for (int i = 0; i < count; i++) {
p1[i] = pre[i+1];
i1[i] = in[i];
}
tree.left = reConstructBinaryTree(p1, i1);
}
//生成根节点右边的树
if ((in.length - 1 - count) != 0) {//判断右边的树存不存在
int[] p2 = new int[in.length - 1 - count];
int[] i2 = new int[in.length - 1 - count];
int right0 = count + 1;//根节点右边的第一个数
for (int i = right0; i < in.length; i++) {
p2[i - right0] = pre[i];
i2[i - right0] = in[i];
}
tree.right = reConstructBinaryTree(p2, i2);
}
return tree;
}
}