继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

剑指Offer-重建二叉树

刘信坚
关注TA
已关注
手记 52
粉丝 7795
获赞 665

        最近一直在复习一些算法及数据结构方面的东西,就找了一个适合找工作笔试的题目,在剑指Offer上刷了几道题目,发现对复习知识点还是很有用的,做到重建二叉树这块。递归传值出了点问题,debug半小时才找出错误,所有还是写篇博客记录一下。也推荐要找工作的伙伴去剑指Offer刷题。

       

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

        一看这题目就假如是手写的话,学过数据结构的都知道很简单,一般都是根据前序和中序求后序,或者根据后序和中序求前序,总之要包含中序,二叉树建好只好,其实后序已经出来了,遍历一遍就好了。

思路分析: 前序遍历的第一个数据肯定是根节点,只要找出后序遍历的索引位置就可以确定当前根节点的左右子树,左右子树又是同一个结构,所以用递归非常简单,下面看代码:

TreeNode .class

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
	@Override
	public String toString() {
		return "TreeNode [val=" + val + ", left=" + left + ", right=" + right + "]";
	}
    
}

 当构造好当前元素之后只需要确定好左右子树即可,主要注意 index-startIn+startPre 这个参数,index-startIn 这是求出了在中序数组的偏移量,再加上startPre就是左子树的pre数组的末元素了,这样就确定好关系了,反复递归就ok了。

	    public static TreeNode reConstructBinaryTree(int [] pre,int [] in) {
	       return reConstructTree(pre, 0 , pre.length -1, in , 0 , in.length -1 );
	    }
	    
	    private static TreeNode reConstructTree(int[] pre, int startPre, int endPre, int[] in, int startIn, int endIn) {
			
	    	for(int index = startIn ; index <= endIn ; index++) {
	    		if(pre[startPre] == in[index]) {
	    			TreeNode treeNode = new TreeNode(in[index]);
	    			treeNode.left = reConstructTree(pre,startPre+1,index-startIn+startPre,in,startIn,index-1);
	    			treeNode.right = reConstructTree(pre,index-startIn+startPre+1,endPre,in,index+1,endIn);
	    			return treeNode;
	    		}
	    	}
	    	return null;
		}

测试:

		private static void foreachBinaryTree(TreeNode treeNode) {
             
			  if(treeNode != null) {
				  foreachBinaryTree(treeNode.left);  
				  foreachBinaryTree(treeNode.right); 
                  System.out.println(treeNode.val);
			  }
		}
	    
		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 binaryTree = reConstructBinaryTree(pre,in);
	    	foreachBinaryTree(binaryTree);
       }

我这里后序遍历了一下二叉树。

7
4
2
5
8
6
3
1




     


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP