你被安排了 n 个任务。任务需要花费的时间用长度为 n 的整数数组 tasks 表示,第 i 个任务需要花费 tasks[i] 小时完成。一个 工作时间段 中,你可以 至多 连续工作 sessionTime 个小时,然后休息一会儿。
你需要按照如下条件完成给定任务:
如果你在某一个时间段开始一个任务,你需要在 同一个 时间段完成它。
完成一个任务后,你可以 立马 开始一个新的任务。
你可以按 任意顺序 完成任务。
给你 tasks 和 sessionTime ,请你按照上述要求,返回完成所有任务所需要的 最少 数目的 工作时间段 。
测试数据保证 sessionTime 大于等于 tasks[i] 中的 最大值 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-number-of-work-sessions-to-finish-the-tasks
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
int minSessions(vector<int>& tasks, int sessionTime) {
int n = tasks.size();
vector<int> valid(1 << n);
// 在[1, 1 << n)范围内枚举每一个mask
// 计算其包含的时间的总和
for (int mask = 1; mask < 1 << n; mask++) {
int needTime = 0;
for (int i = 0; i < n; i++) {
if (mask & (1 << i)) {
needTime += tasks[i];
}
}
if (needTime <= sessionTime) {
valid[mask] = true;
}
}
vector<int> dp(1 << n, 1e9);
dp[0] = 0;
for (int mask = 1; mask < 1 << n; mask++) {
for (int subset = mask; subset; subset = (subset - 1) & mask) {
if (valid[subset]) {
dp[mask] = min(dp[mask], dp[mask ^ subset] + 1);
}
}
}
return dp[(1 << n) - 1];
}
};
用到了一些位运算
枚举 mask 的子集有一个经典的小技巧,对应的伪代码如下:
subset = mask
while subset != 0 do
// subset 是 mask 的一个子集,可以用其进行状态转移
...
// 使用按位与运算在 O(1) 的时间快速得到下一个(即更小的)mask 的子集
subset = (subset - 1) & mask
end while
即上面解答中的for循环。
求解复杂度也很秀,复杂度为pow(3,n)
用到了二项式定理