题目
为了充分发挥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
思路
以示例1为例,每秒任务执行过程如下:
第1秒开始进来1个任务,执行完成后剩余0个任务(1<=3)
第2秒开始进来2个任务,执行完成后剩余0个任务(2<=3)
第3秒开始进来3个任务,执行完成后剩余0个任务(3<=3)
第4秒开始进来4个任务,执行完成后剩余1个任务(4>3)
第5秒开始进来5+1个任务,执行完成后剩余3个任务(6>3)
第6秒开始进来3个任务,执行完成后剩余0个任务(3>=3)
所以结果为6。
使用curNum记录当前进来的任务数量,使用leaveNum记录当前秒结束后的剩余任务数量。
通过上述过程,可以看到结束的标志是:时间大于等于数组长度并且leaveNum为0。
题解
package hwod;
import java.util.Scanner;
public class ExecuteTime {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int maxTask = sc.nextInt();
int n = sc.nextInt();
int[] nums = new int[n];
for (int i = 0; i < n; i++) {
nums[i] = sc.nextInt();
}
System.out.println(executeTime(nums, maxTask));
}
private static int executeTime(int[] nums, int maxTask) {
int res = 0;
int curNum, leaveNum = 0;
while (res < nums.length || leaveNum != 0) {
curNum = leaveNum;
if (res < nums.length) curNum += nums[res];
leaveNum = curNum > maxTask ? curNum - maxTask : 0;
res++;
}
return res;
}
}
推荐
如果你对本系列的其他题目感兴趣,可以参考华为OD机试真题及题解(JAVA),查看当前专栏更新的所有题目。