题解 | #牛群的树形结构重建#

牛群的树形结构重建

https://www.nowcoder.com/practice/bcabc826e1664316b42797aff48e5153?tpId=354&tqId=10591490&ru=/exam/oj&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj%3Fpage%3D1%26tab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D354

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param inOrder int整型一维数组
     * @param postOrder int整型一维数组
     * @return TreeNode类
     */
    public TreeNode buildTree (int[] inOrder, int[] postOrder) {
        // write code here
        return buildTreeHelper(inOrder, postOrder, inOrder.length - 1, 0,
                               inOrder.length - 1);
    }

    private static TreeNode buildTreeHelper(int[] inorder, int[] postorder,
                                            int postIndex, int inStart, int inEnd) {
        if (postIndex < 0 || inStart > inEnd) {
            return null;
        }

        TreeNode root = new TreeNode(postorder[postIndex]);
        int rootIndexInorder = findIndex(inorder, inorder.length, postorder[postIndex]);

        root.right = buildTreeHelper(inorder, postorder, postIndex - 1,
                                     rootIndexInorder + 1, inEnd);
        root.left = buildTreeHelper(inorder, postorder,
                                    postIndex - (inEnd - rootIndexInorder) - 1, inStart, rootIndexInorder - 1);

        return root;
    }
    private static int findIndex(int[] array, int length, int value) {
        for (int i = 0; i < length; i++) {
            if (array[i] == value) {
                return i;
            }
        }
        return -1;
    }
}

知识点:

  1. 基本的Java语法和概念。
  2. 二叉树的构建和遍历。
  3. 递归算法的应用。

解题思路:

在这个实现中,我们首先根据后序遍历数组确定根节点,然后在中序遍历数组中找到根节点的位置,从而将数组分为左子树和右子树部分。然后,我们递归地构建左子树和右子树。

buildTreeHelper方法中,postIndex表示当前在后序遍历数组中的索引,inStart和inEnd表示当前中序遍历数组中的起始和结束索引。我们在中序遍历数组中找到根节点的索引,然后根据该索引将数组分为左右子树,并继续递归地构建这些子树。

最终,我们可以通过调用inorderTraversal方法来验证构建的二叉树是否正确。

全部评论

相关推荐

09-27 18:15
门头沟学院 C++
在努力的小牛:来告诉你 录用评估挂就是同期好几个候选人,部门负责人选了其他人。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务