华为OD机试-- 执行时长、GPU算力

华为OD机试-- 执行时长、GPU算力

问题描述

为了充分发挥GPU算力，需要尽可能多的将任务交给GPU执行，现在有一个任务数组，数组元素表示在这1秒内新增的任务个数且每秒都有新增任务。

假设GPU最多一次执行n个任务，一次执行耗时1秒，在保证GPU不空闲情况下，最少需要多长时间执行完成。

• 输入描述:

  • 第一个参数为GPU一次最多执行的任务个数，取值范围[1, 10000]
  • 第二个参数为任务数组长度，取值范围[1, 10000]
  • 第三个参数为任务数组，数字范围[1, 10000]

• 输出描述:

  • 执行完所有任务最少需要多少秒。

示例1

输入
3
5
1 2 3 4 5
输出
6
说明
一次最多执行3个任务，最少耗时6s

示例2

输入
4
5
5 4 1 1 1
输出
5

解决思路

• 题目中说GPU最多一次执行N个任务，同时也说了任务是有长度的，那就意味着同一时间任务的长度不能大于给定的GPU数量，因此，我们可以对任务数组进行遍历，遍历时记录任务的耗时及剩余GPU的数量，当下一个任务的长度和剩余的GPU数量满足总的时就可以将剩余的GPU归零。

代码实现

public static void main(String[] args) {
    int[] nums = new int[args.length];
    for (int i = 0; i < nums.length; i++) {
        nums[i] = Integer.parseInt(args[i]);
    }
    int gpuNum = nums[0];
    int ans = 0;
    int more = 0;
    for (int i = 2; i < nums.length; i++) {
        int x = nums[i];
        if (x + more > gpuNum) {
            more = more + x - gpuNum;
        } else {
            more = 0;
        }
        ans++;
    }
    while (more > 0) {
        more = more - gpuNum;
        ans++;
    }
    System.out.println(ans);
}