您的位置:首页 > Web前端

剑指offer--重建二叉树

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

分类:二叉树

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

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

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

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

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