【LeetCode】提莫攻击Java题解

255 阅读3分钟

这是我参与11月更文挑战的第10天,活动详情查看:2021最后一次更文挑战

题目描述

在《英雄联盟》的世界中,有一个叫 “提莫” 的英雄。他的攻击可以让敌方英雄艾希(编者注:寒冰射手)进入中毒状态。

当提莫攻击艾希,艾希的中毒状态正好持续 duration 秒。

正式地讲,提莫在 t 发起发起攻击意味着艾希在时间区间 [t, t + duration - 1](含 t 和 t + duration - 1)处于中毒状态。如果提莫在中毒影响结束 前 再次攻击,中毒状态计时器将会 重置 ,在新的攻击之后,中毒影响将会在 duration 秒后结束。

给你一个 非递减 的整数数组 timeSeries ,其中 timeSeries[i] 表示提莫在 timeSeries[i] 秒时对艾希发起攻击,以及一个表示中毒持续时间的整数 duration 。

返回艾希处于中毒状态的 总 秒数。

示例 1:

输入:timeSeries = [1,4], duration = 2
输出:4
解释:提莫攻击对艾希的影响如下:
- 第 1 秒,提莫攻击艾希并使其立即中毒。中毒状态会维持 2 秒,即第 1 秒和第 2 秒。
- 第 4 秒,提莫再次攻击艾希,艾希中毒状态又持续 2 秒,即第 4 秒和第 5 秒。
艾希在第 1、2、4、5 秒处于中毒状态,所以总中毒秒数是 4 。

示例 2:

输入:timeSeries = [1,2], duration = 2
输出:3
解释:提莫攻击对艾希的影响如下:
- 第 1 秒,提莫攻击艾希并使其立即中毒。中毒状态会维持 2 秒,即第 1 秒和第 2 秒。
- 第 2 秒,提莫再次攻击艾希,并重置中毒计时器,艾希中毒状态需要持续 2 秒,即第 2 秒和第 3 秒。
艾希在第 1、2、3 秒处于中毒状态,所以总中毒秒数是 3 。


来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/teemo-attacking
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路分析

  • 今天的算法每日一题是数组类题目,这个题目结合很火的英雄联盟,提莫攻击场景。分析题干,要求的是中毒状态总时间。
  • 首先,我们可以使用模拟的办法,把每一个中毒的时间都枚举出来,采用 hashSet 记录,最终 hashSet 的长度就是中毒状态总时间。这种解法的时间复杂度是O(n * n), 不能通过所有测试用例。
  • 分析题目要求,只需要求解中毒状态总时间,不需要列出所有的中毒时间,因此,我们可以优化求解方法。还有一个重要隐藏条件,时间是递增的,题目中timeSeries是正向递增的。 我们可以遍历timeSeries, 每一个 time + duration 即为中毒时间,如果当前时间 大于 中毒时间,可以直接加上 duration。否则,则需要减去前一个 time 的中毒时间。实现代码如下:

通过代码

  • 朴素解法
class Solution {
    public int findPoisonedDuration(int[] timeSeries, int duration) {
        Set<Integer> set = new HashSet<>();
        for (int time : timeSeries) {
            for (int i = 0; i < duration; i++) {
                set.add(time + i);
            }
        }

        return set.size();
    }
}
  • 优化解法
class Solution {
    public int findPoisonedDuration(int[] timeSeries, int duration) {
        int ans = 0;
        int expired = 0;
        for (int time : timeSeries) {
            if (time > expired) {
                ans += duration;
            } else {
                ans += time + duration - expired;
            }
            
            expired = time + duration;
        }

        return ans;
    }
}

image.png

总结

  • 朴素解法的时间复杂度是O(n * n),空间复杂度是O(n)
  • 优化解法的时间复杂度是O(n),空间复杂度是O(1)
  • 坚持算法每日一题,加油!