你有一个炸弹需要拆除,时间紧迫!你的情报员会给你一个长度为 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
,请你返回解密后的结果来拆除炸弹!
示例 1:
输入:code = [5,7,1,4], k = 3 输出:[12,10,16,13] 解释:每个数字都被接下来 3 个数字之和替换。解密后的密码为 [7+1+4, 1+4+5, 4+5+7, 5+7+1]。注意到数组是循环连接的。
示例 2:
输入:code = [1,2,3,4], k = 0 输出:[0,0,0,0] 解释:当 k 为 0 时,所有数字都被 0 替换。
示例 3:
输入:code = [2,4,9,3], k = -2 输出:[12,5,6,13] 解释:解密后的密码为 [3+9, 2+3, 4+2, 9+4] 。注意到数组是循环连接的。如果 k 是负数,那么和为 之前 的数字。
提示:
n == code.length
1 <= n <= 100
1 <= code[i] <= 100
-(n - 1) <= k <= n - 1
问题简要描述:返回解密后的结果来拆除炸弹
Java
class Solution {public int[] decrypt(int[] code, int k) {int n = code.length;int[] ans = new int[n];if (k == 0) {return ans;}for (int i = 0; i < n; i++) {if (k > 0) {for (int j = i + 1; j < i + k + 1; j++) {ans[i] += code[j % n];}} else if (k < 0) {for (int j = i + k; j < i; j++) {ans[i] += code[(j + n) % n];}}}return ans;}
}
Python3
class Solution:def decrypt(self, code: List[int], k: int) -> List[int]:n = len(code)ans = [0] * nif k == 0:return ansfor i in range(n):if k > 0:for j in range(i + 1, i + k + 1):ans[i] += code[j % n]elif k < 0:for j in range(i + k, i):ans[i] += code[(j + n) % n]return ans
TypeScript
function decrypt(code: number[], k: number): number[] {const n: number = code.length;const ans: number[] = Array(n).fill(0);if (k == 0) {return ans;}for (let i = 0; i < n; i++) {if (k > 0) {for (let j = i + 1; j < i + k + 1; j++) {ans[i] += code[j % n];}} else if (k < 0) {for (let j = i + k; j < i; j++) {ans[i] += code[(j + n) % n];}}}return ans;
};