LeetCode 2365. 任务调度器 II 模拟+哈希

题目描述

给你一个下标从 0 开始的正整数数组 tasks ，表示需要 按顺序 完成的任务，其中 tasks[i] 表示第 i 件任务的 类型 。

同时给你一个正整数 space ，表示一个任务完成 后 ，另一个 相同 类型任务完成前需要间隔的 最少 天数。

在所有任务完成前的每一天，你都必须进行以下两种操作中的一种：

完成 tasks 中的下一个任务
休息一天
请你返回完成所有任务所需的 最少 天数。

示例 1：

输入：tasks = [1,2,1,2,3,1], space = 3
输出：9
解释：
9 天完成所有任务的一种方法是：
第 1 天：完成任务 0 。
第 2 天：完成任务 1 。
第 3 天：休息。
第 4 天：休息。
第 5 天：完成任务 2 。
第 6 天：完成任务 3 。
第 7 天：休息。
第 8 天：完成任务 4 。
第 9 天：完成任务 5 。
可以证明无法少于 9 天完成所有任务。
示例 2：

输入：tasks = [5,8,8,5], space = 2
输出：6
解释：
6 天完成所有任务的一种方法是：
第 1 天：完成任务 0 。
第 2 天：完成任务 1 。
第 3 天：休息。
第 4 天：休息。
第 5 天：完成任务 2 。
第 6 天：完成任务 3 。
可以证明无法少于 6 天完成所有任务。

提示：

1 <= tasks.length <= 105
1 <= tasks[i] <= 109
1 <= space <= tasks.length

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/task-scheduler-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

算法

1. 按顺序遍历任务数组，day记录天数
2. 遍历任务数组，维护每个任务最后完成的时间，如果大于间隔，直接完成，否则day等待。

为了便于编程，上面的条件可以改成如果小于间隔等待。

然后完成。

注意

看数据范围，有可能最后完成的时间会爆int，因此，我之前竟然傻乎乎的去弄了一个时间戳，这明显会超时啊，时间一天天涨，等到什么时候了。明显要遍历任务数组

代码

#define LL long long

class Solution {
public:
    LL taskSchedulerII(vector<int>& tasks, int space) {
        unordered_map<int, LL> last;
        LL day = 0;

        for (int t : tasks) {
            if (last.find(t) != last.end() && day - last[t] <= space)
                day = last[t] + space + 1;

            last[t] = day;
            day++;
        }

        return day;
    }
};

作者：wzc1995
链接：https://www.acwing.com/solution/content/130603/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

这里直接利用了上面的变形，简化了代码的编写。否则要写一堆的判断语句。

		unordered_map<int, long long> last;
        long long day = 0;

        for (int t : tasks) {
            if (last.find(t) == last.end()) {
                last[t] = day ++;
                continue;
            }
            
            if (day - last[t] <= space) {
                day = last[t] + space + 1; 
                last[t] = day ++;
            }
            else {
                last[t] = day ++;
            }
            
        }

        return day;

下面是我之前傻乎乎维护时间戳。

class Solution {
public:
    long long taskSchedulerII(vector<int>& tasks, int space) {
        // int last[100010] = {0};
        int n = tasks.size();
        unordered_map<int, long long> last(n + 1);
        long long time = 1;
        int i = 0;

        while (1) {
            if (last[tasks[i]] == 0 || time - last[tasks[i]] > space){
                last[tasks[i]] = time;
                i ++ ;
                if (i >= n) break;    
            }

            time ++;
        }

        // for (int i = 0; i < 9; i ++)
        //     cout << last[i] << endl;

        return time;
    }
};

Python

class Solution:
    def taskSchedulerII(self, tasks: List[int], space: int) -> int:
        ans, last = 0, {}
        for t in tasks:
            ans += 1  # 完成该任务，天数+1
            if t in last:
                ans = max(ans, last[t] + space + 1)  # 看看是否要间隔 space 天
            last[t] = ans  # 记录上一次完成时间
        return ans

作者：endlesscheng
链接：https://leetcode.cn/problems/task-scheduler-ii/solution/mo-ni-by-endlesscheng-oiuw/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。