1652. 拆炸弹

最新推荐文章于 2024-07-24 22:51:50 发布

霜白sakura

最新推荐文章于 2024-07-24 22:51:50 发布

阅读量68

点赞数

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

本文链接：https://blog.csdn.net/weixin_44458954/article/details/127028628

版权

leetcode 专栏收录该内容

58 篇文章 0 订阅

订阅专栏

题目：1652. 拆炸弹

class Solution {
public:
    vector<int> decrypt(vector<int>& code, int k) {
        int n = code.size();
        if(k == 0){
            vector<int> res(n,0);
            return res;
        }else if(k>0){
            vector<int> res(n,0);
            for(int i=0;i<n;++i){
                int index = i;
                for(int j=0;j<k;++j){
                    index = next(n,index);
                    res[i] += code[index];
                }
            }
            return res;
        }else if(k<0){
            vector<int> res(n,0);
            for(int i=0;i<n;++i){
                int index = i;
                for(int j=0;j<-k;++j){
                    index = prev(n,index);
                    res[i] += code[index];
                }
            }
            return res;
        }
        return code;
    }
    inline int next(int n,int index){
        return index+1 == n?0:index+1;
    }
    inline int prev(int n,int index){
        return index-1 == -1?n-1:index-1;
    }
};