菜单 学习猿地 - LMONKEY

VIP

开通学习猿地VIP

尊享10项VIP特权 持续新增

知识通关挑战

打卡带练!告别无效练习

接私单赚外块

VIP优先接,累计金额超百万

学习猿地私房课免费学

大厂实战课仅对VIP开放

你的一对一导师

每月可免费咨询大牛30次

领取更多软件工程师实用特权

入驻
210
0

7、重建二叉树

原创
05/13 14:22
阅读数 40766

dfs构建二叉树
tricky的点在于:如何确定子树在数组中对应的起点和终点位置
1、首先确认左子树长度 leftlen = inStart - mid;
则 有 preOrder顺序:
preStart + 1 -> preStart + leftlen
preStart + leftlen + 1 -> preEnd

public TreeNode buildTree(int[] preorder, int[] inorder) {
        //[][]
        //[3,1,2][1,3,2]
        //[3,4,5][3,4,5]?
        //[1,2][]?
        // [3,9,20,15,7][9,3,15,20,7]
        int n = preorder.length;
        int m = inorder.length;
        if (n  == 0 || m == 0 || (n != m)){
            return null;
        }
        return dfs (preorder, inorder,0, n -1 , 0 , m - 1);
    }
    public TreeNode dfs(int[] preorder, int[] inorder, int preStart, int preEnd, int inStart, int inEnd){
        if (preStart > preEnd || inStart > inEnd){
            return null;
        }
        int rootVal = preorder[preStart];
        TreeNode root = new TreeNode(rootVal);

        int mid = inStart;
        while (mid <= inEnd && inorder[mid] != rootVal){
            mid++;
        }

        int leftLen = mid - inStart;//1 - 0 = 1

        root.left = dfs(preorder, inorder, preStart + 1, preStart + leftLen, inStart, mid - 1);
        root.right = dfs(preorder, inorder, preStart + leftLen + 1, preEnd, mid + 1, inEnd);
        return root;
    }

发表评论

0/200
210 点赞
0 评论
收藏
为你推荐 换一批