1.题目基本信息

1.1.题目描述

你有一个炸弹需要拆除，时间紧迫！你的情报员会给你一个长度为 n 的 循环 数组 code 以及一个密钥 k 。

为了获得正确的密码，你需要替换掉每一个数字。所有数字会 同时 被替换。

• 如果 k > 0 ，将第 i 个数字用 接下来 k 个数字之和替换。

• 如果 k < 0 ，将第 i 个数字用 之前 k 个数字之和替换。

• 如果 k == 0 ，将第 i 个数字用 0 替换。

由于 code 是循环的， code[n-1] 下一个元素是 code[0] ，且 code[0] 前一个元素是 code[n-1] 。

给你 循环 数组 code 和整数密钥 k ，请你返回解密后的结果来拆除炸弹！

1.2.题目地址

https://leetcode.cn/problems/defuse-the-bomb/description/

2.解题方法

2.1.解题思路

滑动窗口

2.2.解题步骤

第一步，构建维护变量。sum1维护窗口中的元素和；r维护窗口的右开位置；result维护结果变量。

第二步，初始化第一个窗口的元素和。

第三步，滑动滑动窗口；更新sum1、result和r变量。

3.解题代码

python代码

class Solution:
    def decrypt(self, code: List[int], k: int) -> List[int]:
        # 思路：滑动窗口
        n = len(code)
        if k == 0:
            return [0] * n
        # 第一步，构建维护变量。sum1维护窗口中的元素和；r维护窗口的右开位置；result维护结果变量。
        result = [0] * n
        sum1 = 0
        r = k + 1 if k > 0 else n
        # 第二步，初始化第一个窗口的元素和
        k = abs(k)
        sum1 = sum(code[r - k: r])
        result[0] = sum1
        # 第三步，滑动滑动窗口；更新sum1、result和r变量
        for i in range(1, n):
            sum1 += code[r % n] - code[(r - k) % n]
            result[i] = sum1
            r += 1
        return result

4.执行结果