华为算法机试--用户调度问题

题目描述

给定一个正整数组表示待系统执行的任务列表，数组的每一个元素代表一个任务，元素的值表示该任务的类型。请计算执行完所有任务所需的最短时间。
任务执行规则如下:
任务可以按任意顺序执行，且每个任务执行耗时间均为1个时间单位。
两个同类型的任务之间必须有长度为N个单位的冷却时间，比如N为2时，在时间K执行了类型3的任务，那么K+1和K+2两个时间不能执行类型3任务。
系统在任何一个单位时间内都可以执行一个任务，或者等待状态。
说明：数组最大长度为1000，速度最大值1000。

输入描述
第一行记录一个用半角逗号分隔的数组，数组长度不超过1000，数组元素的值不超过1000，
第二行记录任务冷却时间，N为正整数，N<=100。

输出描述
输出为执行完所有任务所需的最短时间。

示例1 ：
输入输出示例仅供调试，后台判题数据一般不包含示例

输入

2,2,2,3
2

输出

7

说明
时间1：执行类型2任务。
时间2：执行类型3的任务（因为冷却时间为2，所以时间2不能执行类型2的任务）。
时间3：系统等待（仍然在类型2的冷却时间）。
时间4：执行类型2任务。
时间5：系统等待。
时间6：系统等待。
时间7：执行类型2任务。
因此总共耗时7。

解题源码： 给出Java和python两个版本供参考
python版本：

def task_schedule():
    arr = input().strip().split(",")
    n = int(input().strip())
    d = {} 
    d_max = 0 
    d_max_n = 0  
    for i in arr:
        d[i] = d.get(i, 0) + 1
        if d[i] > d_max:
            d_max = d[i]
    for k, v in d.items():
        if d[k] == d_max:
            d_max_n += 1
    return max((d_max - 1) * (n + 1) + d_max_n, len(arr))


if __name__ == "__main__":
    print(task_schedule())

Java版本：

public static void main(String[] args) {
        try (Scanner in = new Scanner(System.in)) {
            String[] split = in.nextLine().split(",");
            int N = in.nextInt();
            int[] jobs = Arrays.stream(split).mapToInt(Integer::parseInt).toArray();
            long dMax = 0L;
            long d_max_n = 0L;
            HashMap<Integer, Long> jobMap = new HashMap<>(jobs.length);
            for (int job : jobs) {
                jobMap.put(job, jobMap.getOrDefault(job, 0L) + 1);
                if (jobMap.get(job) > dMax) {
                    dMax = jobMap.get(job);
                }
            }
            for (Integer key : jobMap.keySet()) {
                if (jobMap.get(key) == dMax) {
                    d_max_n += 1;
                }
            }
            long result = Math.max((dMax - 1) * (N + 1) + d_max_n, jobs.length);
            System.out.println(result);
        }
}