输入某二叉树的前序和中序遍历的结果,重建出该二叉树。

重建二叉树

http://www.nowcoder.com/questionTerminal/8a19cbe657394eeaac2f6ea9b0f6fcf6

通过递归就行,关键是每一步的下标范围要对准!最好画草图来确认;
思路如下:
每次先序遍历第一个结点,则该结点为当前子树下的根a,再在中序遍历中找到根a的位置,a的左边是左子树,右边是右子树。
然后在通过相同的方式,一次遍历左右子树即可。

/**
 * 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 geTree(pre, in, 0, pre.length - 1, 0, in.length - 1);
    }

    public TreeNode geTree(int[] pre, int[] in, int l1, int r1, int l2, int r2) {
                //l1,r1分别为子树的根、尾结点的前序的下标。同理。。。 
        if (l1 > r1) {
            return null;
        } else {
            int i;
            for (i = l2; i <= r2; i++) {
                if (pre[l1] == in[i]) {
                    break;
                }
            }
            TreeNode treeNode = new TreeNode(in[i]);
            treeNode.left = geTree(pre, in, l1 + 1, l1 + i - l2, l2, i - 1);
            treeNode.right = geTree(pre, in, l1 + i - l2 + 1, r1, i + 1, r2);
            return treeNode;
        }
    }
}
全部评论

相关推荐

2 收藏 评论
分享
牛客网
牛客企业服务