华为OD笔试机试 - GPU调度/执行时长 (c++ 2024年C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

为了充分发挥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

说明： 一次最多执行4个任务，最少耗时5s

解题思路

简单模拟题，GPU每次执行n个任务，消耗不完的则延迟到后面执行即可。

参考代码

#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

int main() {
    int m, n;
    cin >> m >> n;
    
    int total = 0;
    for (int i = 0; i < n; i++) {
        int tasks;
        cin >> tasks;
        total += tasks;
        total -= m;
        if (total < 0) total = 0;
    }

    cout << n + (total + m - 1) / m << endl;

    return 0;
}