传递悄悄话 - 华为OD机试题

0 9 20 -1 -1 15 7 -1 -1 -1 -1 3 2

38

输入：
0 9 20 -1 -1 15 15 7 -1 -1 -1 -1 3 2

输出：
38

这个题目是一个树的遍历问题，采用**深度优先搜索（DFS）**的方式解决。

遍历时叶子节点最晚到达时间即为答案。

import java.util.Arrays;
import java.util.Scanner;

/**
 * @author code5bug
 */
public class Main {

    static int[] arr;

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        arr = Arrays.stream(scanner.nextLine().split(" "))
                .mapToInt(Integer::parseInt).toArray();

        System.out.println(dfs(0));
    }

    // 从 idx 节点到叶子节点最大耗时
    static int dfs(int idx) {
        int maxCostTime = 0;

        // 左右节点
        int leftIdx = 2 * idx + 1, rightIdx = 2 * idx + 2;
        if (leftIdx < arr.length && arr[leftIdx] != -1) {
            maxCostTime = Math.max(maxCostTime, dfs(leftIdx));
        }

        if (rightIdx < arr.length && arr[rightIdx] != -1) {
            maxCostTime = Math.max(maxCostTime, dfs(rightIdx));
        }

        return arr[idx] + maxCostTime;
    }
}