Leetcode每日一题_1652. 拆炸弹

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

Shmilysw

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

阅读量150

点赞数

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

本文链接：https://blog.csdn.net/Shmilysw/article/details/127022367

版权

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

1 篇文章 0 订阅

订阅专栏

1652. 拆炸弹

（前缀和）

// 前缀和
class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        int n = code.size();
        vector<int> res(n, 0), t;
        if (k == 0) return res;
        t = code;
        t.insert(t.end(), code.begin(), code.end());
        vector<int> s(2 * n + 1);
        for (int i = 1; i <= 2 * n ; i ++ ) s[i] = t[i - 1] + s[i - 1];
        if (k > 0) {
            for (int i = 1; i <= n ; i ++ ) {
                res[i - 1] = s[i + k] - s[i];
            }
        } else {
            for (int i = 1; i <= n ; i ++ ) {
                res[i - 1] = s[i + n - 1] - s[i + n + k - 1]; // 推导
            }
        }
        return res;
    }
};