牛客网——剑指offer——二叉树——重建二叉树

思路

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;

    }
}