1652. 拆炸弹

最新推荐文章于 2024-06-09 20:18:37 发布

咔咔咔的

最新推荐文章于 2024-06-09 20:18:37 发布

阅读量513

点赞数 7

分类专栏： leetcode 文章标签： c++

本文链接：https://blog.csdn.net/weixin_45256307/article/details/139057289

版权

leetcode 专栏收录该内容

425 篇文章 1 订阅

订阅专栏

1652. 拆炸弹

题目链接：1652. 拆炸弹

代码如下：

class Solution 
{
public:
    vector<int> decrypt(vector<int>& code, int k) 
    {
        vector<int> res(code.size(),0);
        if(k==0)    return res;   
        int n=code.size();
        if(k>0)
        {
            for(int i=0;i<code.size();i++)
            {
                int tmpk=k;
                while(tmpk>0)
                {
                    res[i]+=code[(i+tmpk)%n];
                    tmpk--;
                }
            }
        }
        else if(k<0)
        {
            for(int i=0;i<code.size();i++)
            {
                int tmpk=k;
                while(tmpk<0)
                {
                    int pos=i+tmpk+n;
                    if(pos>=n)  pos%=n;
                    res[i]+=code[pos];
                    tmpk++;
                }  
            }
        }
        return res;
    }
};