华为OD机试真题 Java 实现【查找树中元素】【2023 B卷 100分】，附详细解题思路

news2024/11/17 1:40:18

在这里插入图片描述

一、题目描述

已知树形结构的所有节点信息，现要求根据输入坐标（x,y）找到该节点保存的内容值；

其中：

x表示节点所在的层数，根节点位于第0层，根节点的子节点位于第1层，依次类推；

y表示节点在该层内的相对偏移，从左至右，第一个节点偏移0，第二个节点偏移1，依次类推；

二、输入描述

每个节点以一维数组（int[]）表示，所有节点信息构成二维数组（int[][]），二维数组的0位置存放根节点；

表示单节点的一维数组中，0位置保存内容值，后续位置保存子节点在二维数组中的索引位置；

根节点的可以表示为{10,1,2}，树的整体表示为

{{10,1,2},{-21,3,4},{23,5},{14},{35},{66}}

查询条件以长度为2的一维数组表示，上图查询坐标为(1,1)时表示为

{1,1}

使用Java标准IO键盘输入进行录入时，先录入节点数量，然后逐行录入节点，最后录入查询的位置，对于上述示例为：

10 1 2

-21 3 4

23 5

1 1

三、输出描述

查询到内容值时，输出{内容值}，查询不到时输出{}

根据坐标(1,1)查询输出{23}，根据坐标(1,2}查询输出{}

四、补充说明

考试者不需要自己编写解析输入文本的代码，请直接使用上述代码中的Parser类解析输入文本；

五、Java算法源码

odes, xy[0], xy[1]);
    System.out.println(result);
}

private static int[] parseOneLine(String text) {
    ByteArrayInputStream stream = new ByteArrayInputStream(text.getBytes(StandardCharsets.UTF_8));
    Scanner in = new Scanner(stream);
    List<Integer> list = new ArrayList<>();
    while (in.hasNext()) {
        list.add(in.nextInt());
    }
    return list.stream().mapToInt(it -> it).toArray();
}

private static String doQuery(int[][] nodes, int x, int y) {
    if (x < 0 || y < 0) {
        return "{}";
    }

    List<Integer> list = new ArrayList<>();
    handle(nodes, 0, x, list);

    if (y >= list.size()) {
        return "{}";
    }

    return "{" + list.get(y) + "}";
}

private static void handle(int[][] nodes, int index, int n, List<Integer> list) {
    int[] node = nodes[index];
    if (n == 0) {
        list.add(node[0]);
        return;
    }

    if (node.length == 1) {
        return;
    }

    for (int i = 1; i < node.length; i++) {
        handle(nodes, node[i], n - 1, list);
    }
}