华为OD E卷（100分）21-传递悄悄话

arnold662024-12-16 13:29

前言

工作了十几年，从普通的研发工程师一路成长为研发经理、研发总监。临近40岁，本想辞职后换一个相对稳定的工作环境一直干到老, 没想到离职后三个多月了还没找到工作，愁肠百结。为了让自己有点事情做，也算提高一下自己的编程能力，无聊之余打算用一些大厂的编程题练练手。希望通过这些分享能够帮到一些人，也希望能和看到此文的大神们沟通交流，提升自己，更希望在此期间能够找到一份理想的工作。

题目描述

给定一个二叉树，每个节点上站一个人，节点数字表示父节点到该节点传递悄悄话需要花费的时间。

初始时，根节点所在位置的人有一个悄悄话想要传递给其他人，求二叉树所有节点上的人都接收到悄悄话花费的时间。

输入

给定二叉树

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

注： -1 表示空节点

输出

返回所有节点都接收到悄悄话花费的时间

示例

示例1

输入：

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

38

解题思路

使用深度优先搜索 (DFS) 来解决此问题，递归遍历整棵树，计算从根节点传递到最远叶子节点的时间。

题解

Java实现

java 复制代码

package huawei.e100;

import java.util.Scanner;

/**
* @author arnold
* @date 2024年12月14日
*/
public class T21 {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		while(sc.hasNext()) {
			String[] tmp = sc.nextLine().split(" ");
			int[] data = new int[tmp.length];
			// 最大消耗时间即为传递时间
			int maxTime = getTime(data, 0);
			System.out.println(maxTime);
		}

	}
	
	static int getTime(int[] data, int index){
		if(index >= data.length) {
			return 0;
		}
		// 二叉树左支 坐标 index*2+1 
		int left = getTime(data, index*2 + 1);
		// 二叉树右支 坐标 index*2+2
		int right = getTime(data, index*2 + 2);
		
		// 通过递归获取最大值
		int max = Math.max(left, right);
		
		return data[index] + max;
	}

}