6.重建二叉树

来源:互联网 发布:grub 添加ubuntu引导 编辑:程序博客网 时间:2024/06/14 11:15

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

对于给的样例,树的结构应该为

                   1              /          \            2             3           /           /       \          4           5         6           \                   /            7                 8

先序遍历和中序遍历的规律为:先序遍历的第一个节点为根( 1),中序遍历中找到该节点(即 1),在中序遍历中,在1的左边的为1 的左子树,在2 的右边的为右子树。
在左子树中找(4,7,2)。然后在前序遍历的再往后找到2(后面的第一个),在中序遍历中找到2,在2的左边的(4,7)为左子树,右边的( 空)为右子树。
在左子树中找(4,7)。然后在前序遍历中再往后找到4(后面的第一个),在中序遍历中找到4,在4的左边(空)的为左子树,右边的(7)为右子树。
在左子树中找(7)。然后在前序遍历中再往后找到7(后面的第一个),在中序遍历中找到7,在4的左边(空)的为左子树,右边的(空)为右子树。
结束。
到右子树,和左边相似。
递归的代码为:

public class Solution {    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {        TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);        return root;    }    private TreeNode reConstructBinaryTree(int[] pre, int prestart,int preend,int[] in,int instart, int inend) {        if (prestart>preend ||instart>inend) {            return null;        }        TreeNode root = new TreeNode(pre[prestart]);        for (int i = 0; i <=preend ; i++) {            if (in[i] == pre[prestart]) {                root.left = reConstructBinaryTree(pre,prestart+1,i-instart+prestart,in,instart,i);                root.right = reConstructBinaryTree(pre,i-instart+prestart+1,preend,in,i+1,inend);            }        }        return root;    }}
原创粉丝点击