题目描述
给你一个下标从 0 开始的整数数组 tasks ,其中 tasks[i] 表示任务的难度级别。在每一轮中,你可以完成 2 个或者 3 个 相同难度级别 的任务。
返回完成所有任务需要的 最少 轮数,如果无法完成所有任务,返回 -1
。
示例 1:
输入:tasks = [2,2,3,3,2,4,4,4,4,4]
输出:4
解释:要想完成所有任务,一个可能的计划是:
- 第一轮,完成难度级别为 2 的 3 个任务。
- 第二轮,完成难度级别为 3 的 2 个任务。
- 第三轮,完成难度级别为 4 的 3 个任务。
- 第四轮,完成难度级别为 4 的 2 个任务。
可以证明,无法在少于 4 轮的情况下完成所有任务,所以答案为 4 。
示例 2:
输入:tasks = [2,3,3]
输出:-1
解释:难度级别为 2 的任务只有 1 个,但每一轮执行中,只能选择完成 2 个或者 3 个相同难度级别的任务。因此,无法完成所有任务,答案为 -1 。
提示:
- 1 <= tasks.length <= 105
- 1 <= tasks[i] <= 109
题解思路
首先统计不同难度级别的任务各自出现的频率,然后对频率( ≥1)进行分类讨论:
- 频率是 1,说明这种任务无法完成。
- 频率是 3×k,k 为 ≥1 的整数。每次完成 3 个,k 轮完成。
- 频率是 3×k+2,k 为 ≥0 的整数。其中 3×k 个任务需要 k 轮完成,剩下 2 个任务需要 1 轮完成。
- 频率是 3×k+1,k 为 ≥1 的整数。其中 3×(k−1) 个任务需要 (k−1) 轮完成,剩下 4 个任务需要 2 轮完成。
对这些情况进行求和即可。
代码实现:
class Solution {
public:
int minimumRounds(vector<int>& tasks) {
unordered_map<int, int> cnt;
for (int i : tasks) {
cnt[i]++;
}
int res = 0;
for (auto num: cnt){
if (num.second == 1) {
return -1;
}
else if(num.second % 3 == 0) {
res += num.second/3;
}
else {
res += num.second/3 + 1;
}
}
return res;
}
};