Defuse the Bomb.cpp

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