Defuse the Bomb LeetCode solution in C++

1
class Solution {
2
 public:
3
  vector<int> decrypt(vector<int> &code, int k) {
4
    vector<int> Ans(code.size(), 0);
5

6
    for (int i = 0; i < code.size(); i++) {
7
      if (k > 0) {
8
        int total = 0;
9
        for (int count = 1; count <= k; count++) {
10
          int num = i + count;
11
          num = num % code.size();
12
          total += code[num];
13
        }
14
        Ans[i] = total;
15
      } else if (k < 0) {
16
        int total = 0;
17
        for (int count = -1; count >= k; count--) {
18
          int num = i + count;
19
          if (num < 0) {
20
            num = code.size() + num;
21
          }
22
          total += code[num];
23
        }
24
        Ans[i] = total;
25
      } else {
26
        Ans[i] = 0;
27
      }
28
    }
29
    return Ans;
30
  }
31
};