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

其实对于二叉树的各种遍历 算法的具体实现难度远远大于理解,对我来说在上 数据结构课时,就对前、中、后三种遍历思想有自认为还可以的理解(也就是说给我一棵树,我绝对可以准确的写出它的三种遍历序列,也可通过已知的前(后)序列和中序序列重新画出这棵树),但到了具体的代码实现就比较懵逼了。。。对递归这种需要在脑子中像计算机一样跑N遍的方式好难准确无误的推敲啊。。。每次遇到递归的算法就很尴尬,看的懂但是不会自己写,导致自己的动态规划算法学的很烂0.0,所以对于递归还得加强学习啊!!!

牛客网剑指offer-重建二叉树_第1张图片

 

如果理解了递归的访问,那么建树的过程就容易多了,前序遍历序列的第一个数(后序遍历的最后一个数)一定是根结点,所以可以根据此结点在中序序列中的位置把中序序列分为左子树和右子数两个部分,同样的道理,在左子树和右子数中同样可以用到这样的规律来确定每个中间结点。

牛客网剑指offer-重建二叉树_第2张图片

具体代码实现


package www.niuke.offer;

/*
 * 输入某二叉树的前序遍历和中序遍历的结果,
 * 请重建出该二叉树
 * 假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
 * 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},
 * 则重建二叉树并返回。
 */
public class test05 {
	public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
		TreeNode root = new TreeNode(pre[0]);// 前序的第一个数定为根
		int len = pre.length;
		// 当只有一个数的时候
		if (len == 1) {
			root.left = null;
			root.right = null;
			return root;
		}
		// 找到中序中的根位置
		int rootval = root.val;
		int i;
		for (i = 0; i < len; i++) {
			if (rootval == in[i])
				break;
		}
		// 创建左子树
		if (i > 0) {
			int[] pr = new int[i];
			int[] ino = new int[i];
			for (int j = 0; j < i; j++) {
				pr[j] = pre[j + 1];
			}
			for (int j = 0; j < i; j++) {
				ino[j] = in[j];
			}
			root.left = reConstructBinaryTree(pr, ino);
		} else {
			root.left = null;
		}
		// 创建右子树
		if (len - i - 1 > 0) {
			int[] pr = new int[len - i - 1];
			int[] ino = new int[len - i - 1];
			for (int j = i + 1; j < len; j++) {
				ino[j - i - 1] = in[j];
				pr[j - i - 1] = pre[j];
			}
			root.right = reConstructBinaryTree(pr, ino);
		} else {
			root.right = null;
		}

		return root;
	}

	public static void main(String[] args) {
		int[] pre = { 1, 2, 4, 7, 3, 5, 6, 8 };
		int[] in = { 4, 7, 2, 1, 5, 3, 8, 6 };
		TreeNode t = new test05().reConstructBinaryTree(pre, in);
	}
}

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

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






你可能感兴趣的:(剑指offer)