Defuse the Bomb LeetCode solution in JavaScript

1
/**
2
 * @param {number[]} code
3
 * @param {number} k
4
 * @return {number[]}
5
 */
6
var decrypt = function (code, k) {
7
  var res = new Array(code.length).fill(0);
8
  if (k > 0) {
9
    for (var i = 0; i < code.length; i++) {
10
      var count = 0;
11
      var j = i + 1;
12
      while (count < k) {
13
        if (j === code.length) j = 0;
14
        res[i] += code[j];
15
        count = count + 1;
16
        j++;
17
      }
18
    }
19
  }
20
  if (k < 0) {
21
    for (var i = 0; i < code.length; i++) {
22
      var count = 0;
23
      var j = i - 1;
24
      while (count > k) {
25
        if (j === -1) j = code.length - 1;
26
        res[i] += code[j];
27
        count = count - 1;
28
        j--;
29
      }
30
    }
31
  }
32
  return res;
33
};