【dp】1986. 完成任务的最少工作时间段

你被安排了 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)
用到了二项式定理

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值