华为OD机试 -最小叶子节点(Java)

题目描述

二叉树也可以用数组来存储，
给定一个数组，树的根节点的值储存在下标1，
对于储存在下标n的节点，他的左子节点和右子节点分别储存在下标2n和2n+1，
并且我们用-1代表一个节点为空，
给定一个数组存储的二叉树，
试求从根节点到最小的叶子节点的路径，
路径由节点的值组成。

输入描述

输入一行为数组的内容，
数组的每个元素都是正整数，元素间用空格分割，
注意第一个元素即为根节点的值，
即数组的第n元素对应下标n，
下标0在树的表示中没有使用，所以我们省略了，
输入的树最多为7层。

输出描述

输出从根节点到最小叶子节点的路径上各个节点的值，
由空格分割，
用例保证最小叶子节点只有一个。

示例一

输入

3 5 7 -1 -1 2 4

输出

3 7 2

示例二

输入

5 9 8 -1 -1 7 -1 -1 -1 -1 -1 6

输出

5 8 7 6

详细代码

import org.apache.commons.lang3.StringUtils;

import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

/**
 * @Author: gyt
 * @Version: 1.0
 **/
public class Main {
    private static List<Integer> res = new LinkedList<>();

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String str = scanner.nextLine();
        solution(str);
    }

    private static void solution(String str) {
        // 补充下标为0的元素-1
        str = "-1 " + str;
        String[] split = str.split(" ");
        List<Integer> list = Arrays.stream(split)
                .map(Integer::parseInt).collect(Collectors.toList());
        int min = Integer.MAX_VALUE;
        int minPosition = 0;
        for (int i = 2; i < list.size(); i++) {
            int temp = list.get(i);
            if (temp != -1 && temp < min) {
                min = temp;
                minPosition = i;
            }
        }
        recursion(minPosition, list);
        // 反转
        Collections.reverse(res);
        String result = StringUtils.join(res, " ");
        System.out.println(result);

    }

    private static void recursion(int minPos, List<Integer> list) {
        res.add(list.get(minPos));
        if (minPos == 1) {
            return;
        }
        int tempPos = minPos % 2 == 0 ? minPos / 2 : (minPos - 1) / 2;
        recursion(tempPos, list);
    }
}

运行结果