Leetcode 每日一题 2022.09.24 1652. 拆炸弹

最新推荐文章于 2024-06-15 09:23:13 发布

same-pxt

最新推荐文章于 2024-06-15 09:23:13 发布

阅读量78

点赞数

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

本文链接：https://blog.csdn.net/m0_61509617/article/details/127022991

版权

leetcoed每日一题监督学习专栏收录该内容

17 篇文章 0 订阅

订阅专栏

1652. 拆炸弹

在这里插入图片描述
这道题就是水题一道，只要根据k的情况分类讨论，向前数k个数的时候遇到0时从数组尾部开始，遇到数组尾部使用 % 取到前部分即可。需要注意的是不能在原数组操作，所求的是改变前的和，所以需要用另一个数组返回答案。

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