您的位置:首页 > Web前端

剑指offer--重建二叉树

2017-08-07 16:16 399 查看
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。

分类:二叉树

解法1:明白由先序遍历和中序遍历生成二叉树的过程就好了。首先找到先序遍历的第一个节点,就是根节点,然后在中序遍历中找到根节点。

此时,根节点左右两边,分别是左右子树的中序遍历。

再对于先序遍历,从第二个节点开始,到中序遍历中左子树的长度,就是左子树的先序遍历。

同理,右子树也是一样,这是一个递归过程。

[java] view
plain copy

/** 

 * 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) {  

        return build(pre,in,0,pre.length-1,0,in.length-1);          

    }  

      

    public TreeNode build(int[] pre,int[] in,int pstart,int pend,int istart,int iend){  

        if(pstart>pend) return null;          

        int cur = pre[pstart];//根节点  

        int find = istart;  

        while(find<=iend){//在中序遍历中查找根节点  

            if(cur==in[find]) break;  

            else find++;  

        }  

        int len = find-istart;  

        TreeNode res = new TreeNode(cur);//创建根节点  

        res.left = build(pre,in,pstart+1,pstart+len,istart,find-1);//创建左子树  

        res.right = build(pre,in,pstart+len+1,pend,find+1,iend);//创建右子树  

        return res;  

    }  

}  
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: