流利说编程题

import java.util.*;

public class TopDownLeftRight {
    public static class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }
    }

    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        if (root == null) {
            return null;
        }
        ArrayList<Integer> result = new ArrayList<>();
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.peek();
            result.add(node.val);
            if (node.left != null) {
                queue.add(node.left);
            }
            if (node.right != null) {
                queue.add(node.right);
            }
            queue.poll();
        }
        return result;
    }

    public TreeNode GenerateTree(int[] arr, int start, int end) {
        if (start == end) {
            return new TreeNode(arr[start]);
        }
        if (start > end) {
            return null;
        }
        TreeNode root = new TreeNode(arr[start]);
        int i = start + 1;
        for (; i < end; i++) {
            if (arr[i] > arr[start]) {
                break;
            }
        }
        if (i == arr.length - 1) {
            root.left = GenerateTree(arr, start + 1, i);
        } else {
            root.left = GenerateTree(arr, start + 1, i - 1);
        }
        if (i <= end && arr[i] > arr[start]) {
        root.right = GenerateTree(arr, i, end);
        }
        return root;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int NodeNum = sc.nextInt();
        int[] NodeArr = new int[NodeNum];
        for (int i = 0; i < NodeNum; i++) {
            NodeArr[i] = sc.nextInt();
        }
        TopDownLeftRight topDownLeftRight = new TopDownLeftRight();
        TreeNode root = topDownLeftRight.GenerateTree(NodeArr, 0, NodeNum - 1);
        ArrayList<Integer> re = topDownLeftRight.PrintFromTopToBottom(root);
        for (int i = 0; i < re.size(); i++) {
            System.out.print(re.get(i));
            System.out.print(" ");
        }
    }
}