华为OD机试(C卷+D卷)2024真题目录(Java & c++ & python)
题目描述
幼儿园组织活动,老师布置了一个任务:
每个小朋友去了解与自己同一个小区的小朋友还有几个。
我们将这些数量汇总到数组 garden 中。
请根据这些小朋友给出的信息,计算班级小朋友至少有几个?
输入描述
输入:garden[] = {2, 2, 3}
garden 数组长度最大为 999
每个小区的小朋友数量最多 1000 人,也就是 garden[i] 的范围为 [0, 999]
输出描述
输出:7
用例
输入
2 2 3
输出
7
说明
第一个小朋友反馈有两个小朋友和自己同一小区,即此小区有3个小朋友。 第二个小朋友反馈有两个小朋友和自己同一小区,即此小区有3个小朋友。
这两个小朋友,可能是同一小区的,且此小区的小朋友只有3个人。
第三个小区反馈还有3个小朋友与自己同一小区,则这些小朋友只能是另外一个小区的。这个小区有4个小朋友。
解题思路
假设:
- 小朋友A反馈有x个人与自己的小区相同
- 小朋友B反馈有y个人与自己的小区相同
那么,在什么情况下,可以假设小朋友A和小朋友B是同一个小区的?
答案:只有当x == y时,A和B的小区才可能是相同的。
比如:
小朋友A反馈有1个人和自己小区相同,小朋友B反馈有2个人和自己小区相同。
那么此时小朋友A和小朋友B的小区必然不同。
比如:
小朋友A反馈有1个人和自己小区相同,小朋友B也反馈有1个人和自己小区相同。
那么此时小朋友A,B可能是同一个小区的。并且可以进一步假设该小区只有A,B两个小朋友,即我们可以认为A,B发生了合并。
假设:有x个小朋友都反馈有y个人与自己小区相同
那么此时“至多”有 x * (y+1) 个小朋友,即这x个小朋友的小区各不相同,而这x个小朋友的每一个人都反馈还有y个人的小区和自己相同,即每一个小区都有 y + 1 人。
那么“至少”情况该如何分析呢?此时需要利用到前面的“合并”操作。
由于这x个小朋友都各自反馈有y个人和自己小区相同。因此我们可以将这x个小朋友,分成多段,每段y+1个小朋友,而每段的这些小朋友是可以合并的,即最终会有:ceil( x / (y + 1) ) * (y+1)
其中 ceil( x / (y+1) ) 就是分段数,这里向上取整的原因是,对于某一段不足y+1个小朋友时,
比如有4个小朋友都反馈有2个人和自己小区相同,那么这4个小朋友中,可以选择3人组成一段,剩余的4-3=1人单独组成一段,如下所示,3人组成一段的小区都是A,而单独组成一段的小区是B
A A A | B
其中 A A A 三个小朋友互相弥补,组成一段,因此不需要外人加入。而B小朋友说还有2个人和自己小区一样,而当前进行反馈的小朋友已经组合完了,因此需要额外加入2个外人。
A A A | B B B
根据反馈相同小区y的小朋友的数量x,求出至少的分段数,而每个分段就是y+1人。
参考代码
import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
try {
// 读取输入并转换为整数数组
int[] nums = Arrays.stream(sc.nextLine().split(" "))
.mapToInt(Integer::parseInt)
.toArray();
System.out.println(calculateMinimumChickens(nums));
} catch (Exception e) {
System.out.println("0");
}
}
public static int calculateMinimumChickens(int[] nums) {
// num是反馈,feedbackCounts[num]是给出相同反馈的(小朋友)数量
HashMap<Integer, Integer> feedbackCounts = new HashMap<>();
// 统计每个反馈数量
for (int num : nums) {
feedbackCounts.put(num, feedbackCounts.getOrDefault(num, 0) + 1);
}
// 记录结果
int totalChickens = 0;
for (int feedback : feedbackCounts.keySet()) {
int count = feedbackCounts.get(feedback);
int groupSize = feedback + 1;
totalChickens += Math.ceil(count * 1.0 / groupSize) * groupSize;
}
return totalChickens;
}
}