LeetCode_6174_任务调度器Ⅱ

题目链接

• https://leetcode.cn/problems/task-scheduler-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 <= 10^5
• 1 <= tasks[i] <= 10^9
• 1 <= space <= tasks.length

解题思路

模拟法

• 一次遍历，用一个HashMap存储每个任务的上一次时间，遇到时间到现在不够的话就等待到刚好够的时候继续
• 记录该类型任务上一次完成的时间

AC代码

class Solution {
    public long taskSchedulerII(int[] tasks, int space) {
        long ans = 0;
        Map<Integer, Long> map = new HashMap<>();
        for (int i = 0; i < tasks.length; i++) {
            int key = tasks[i];
            ans++;
            if (map.containsKey(key)) {
                ans = Math.max(ans, map.get(key) + space + 1);
            }
            map.put(key, ans);
        }
        return ans;
    }
}