滑动窗口，LeetCode 1652. 拆炸弹

最新推荐文章于 2024-07-22 16:27:42 发布

EQUINOX1

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量356

点赞数 3

分类专栏： leetcode每日一题文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/EQUINOX1/article/details/138460764

版权

leetcode每日一题专栏收录该内容

144 篇文章 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 ，请你返回解密后的结果来拆除炸弹！

2、接口描述

python3

class Solution:
    def decrypt(self, code: List[int], k: int) -> List[int]:

cpp

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {

    }
};

3、原题链接

1652. 拆炸弹

二、解题报告

1、思路分析

对于环形问题往往只需要对下标取模就可以解决

我们考虑维护长度为k的滑窗不断滑动，维护滑窗内的和，完成n个元素的计算

2、复杂度

时间复杂度： O(n)空间复杂度：O(1)

3、代码详解

python3

class Solution:
    def decrypt(self, code: List[int], k: int) -> List[int]:
        n = len(code)
        r = k + 1 if k > 0 else n
        k = abs(k)
        s = sum(code[r - k: r])
        ret = [0] * n
        for i in range(n):
            ret[i] = s
            s += code[r % n] - code[(r - k) % n]
            r += 1
        return ret

cpp

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        int n = code.size();
        int r = k > 0 ? k + 1 : n;
        k = abs(k);
        int s = accumulate(code.begin() + r - k, code.begin() + r, 0);
        vector<int> ret(n);
        for(int i = 0; i < n; i ++){
            ret[i] = s;
            s += code[r % n] - code[(r - k) % n];
            r ++;
        }
        return ret;
    }
};