题目名称

你有一个炸弹需要拆除，时间紧迫！你的情报员会给你一个长度为 n 的 循环 数组 code 以及一个密钥 k 。

为了获得正确的密码，你需要替换掉每一个数字。所有数字会 同时 被替换。

如果 k > 0 ，将第 i 个数字用 接下来 k 个数字之和替换。
如果 k < 0 ，将第 i 个数字用 之前 k 个数字之和替换。
如果 k == 0 ，将第 i 个数字用 0 替换。
由于 code 是循环的， code[n-1] 下一个元素是 code[0] ，且 code[0] 前一个元素是 code[n-1] 。

给你 循环 数组 code 和整数密钥 k ，请你返回解密后的结果来拆除炸弹！

示例

输入：code = [5,7,1,4], k = 3
输出：[12,10,16,13]
解释：每个数字都被接下来 3 个数字之和替换。解密后的密码为 [7+1+4, 1+4+5, 4+5+7, 5+7+1]。注意到数组是循环连接的。

题解

可以使用 求余的方法来求出当前序号前后k位的值

答案

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/**
 * @param {number[]} code
 * @param {number} k
 * @return {number[]}
 */
 var decrypt = function(code, k) {
  if(k === 0) return code.map(item => 0)
  const code1 = [...code], length = code.length
  if(k > 0) {
    return code1.map((item, index) => {
      let total = 0
      for(let i = index + 1; i <= k+index; i++) {
        total += code[i % length]
      }
      return total
    })
  }
  if(k < 0) {
    return code1.map((item, index) => {
      let total = 0
      for(let i = index - 1; i >= index + k; i--) {
        total += code[(length + i) % length]
      }
      return total
    })
  }
};