题目描述
二叉树也可以用数组来存储,
给定一个数组,树的根节点的值储存在下标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);
}
}
运行结果