Leetcode—1652. 拆炸弹【简单】

源代码•宸

已于 2024-05-05 10:15:29 修改

阅读量331

点赞数 1

分类专栏： LeetCode刷题文章标签： leetcode 算法职场和发展 c++ 经验分享

于 2024-05-05 09:39:35 首次发布

本文链接：https://blog.csdn.net/qq_44631615/article/details/138460191

版权

LeetCode刷题专栏收录该内容

398 篇文章 5 订阅

订阅专栏

2024每日刷题（127）

Leetcode—1652. 拆炸弹

在这里插入图片描述

实现代码

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        int codeSize = code.size();
        vector<int> ans(codeSize, 0);
        if(k == 0) {
            return ans;
        }
        
        if(k > 0) {
            for(int i = 0; i < codeSize; i++) {
                for(int j = 1; j <= k; j++) {
                    ans[i] += code[(i + j) % codeSize];
                }
            }
        } else {
            for(int i = 0; i < codeSize; i++) {
                for(int j = -1; j >= k; j--) {
                    ans[i] += code[(i + j + codeSize) % codeSize];
                }
            }
        }
        return ans;
    }
};

运行结果

在这里插入图片描述

滑动窗口实现代码

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        int n = code.size();
        vector<int> ans(n);
        if(k == 0) {
            return ans;
        }

        int sum = 0;
        int start = k > 0 ? 1: n + k;
        int end = k > 0 ? k: n - 1;

        for(int i = start; i <= end; i++) {
            sum += code[i];
        }

        for(int i = 0; i < n; i++) {
            ans[i] = sum;
            sum -= code[start++ % n];
            sum += code[++end % n];
        }
        return ans;
    }
};