拆弹专家指南：轻松驾驭 LeetCode 1652

后端

2023-09-04 02:10:48

在 LeetCode 1652 中，你面临的任务是拆除一枚定时炸弹。时间紧迫，你必须迅速行动！这个问题的核心在于如何高效地计算出执行特定操作所需的时间。本文将详细介绍如何利用前缀和来解决这一问题。

任务简述

你的情报员提供了循环数组 code，其中每个元素 code[i] 表示执行第 i 个操作所需的时间。你必须在恰当的时间执行操作，否则炸弹会爆炸。具体来说，你需要找到一个操作 k，使得执行该操作所需的时间最少。

前缀和登场

拆除炸弹的关键在于巧妙运用前缀和。前缀和数组 prefix[i] 定义为前 i 个元素之和，即：prefix[i] = code[1] + code[2] + ... + code[i]。

巧用前缀和

有了前缀和，你可以轻松确定执行特定操作所需的总时间：

如果操作 i 在前 j 个操作中执行（即 j < i），则所需时间为 prefix[i] - prefix[j]。
如果操作 i 在前 j 个操作后执行（即 j ≥ i），则所需时间为 prefix[i] + (sum - prefix[j])，其中 sum 是循环数组 code 的总和。

实战演练

现在，让我们通过一个例子来加深理解：

给定数组 code = [5, 10, 15, 20]，假设你要执行操作 3。

使用前缀和数组 prefix = [5, 15, 30, 50]。
操作 3 在前 2 个操作中执行，因此所需时间为 prefix[3] - prefix[2] = 30 - 15 = 15。

掌握要点

巧妙运用前缀和计算执行操作所需时间。
根据操作发生的时间（前 j 个操作中或后）选择正确的公式。
始终将循环数组的总和 sum 考虑在内。

代码实现

以下是 Python 代码示例，展示了如何利用前缀和来解决 LeetCode 1652 问题：

def 拆除炸弹(code, k):
    """
    :type code: List[int]
    :type k: int
    :rtype: int
    """
    n = len(code)
    sum_ = sum(code)
    prefix = [0] * (n + 1)
    for i in range(1, n + 1):
        prefix[i] = prefix[i - 1] + code[i - 1]

    min_time = float('inf')
    for i in range(n):
        if k >= i:
            min_time = min(min_time, prefix[i] + (sum_ - prefix[k - i]))
        else:
            min_time = min(min_time, prefix[i] - prefix[k])

    return min_time