1.问题描述:
你有一个炸弹需要拆除,时间紧迫!你的情报员会给你一个长度为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/defuse-the-bomb
2. 思路分析:
分析题目可以知道我们需要根据k的值分为三种情况讨论,当k == 0时直接返回一个长度为len(code)的数组,当k > 0时,因为是需要计算出当前位置之后的k个元素和,所以我们可以一开始的时候先计算出第一个位置后面的k个元素和sum,然后通过通过一个循环计算剩余的情况,sum减去当前位置下一个位置的code数组对应的值,再加上(i + k + 1) % n的code数组对应的值,我们可以在循环的时候将计算的结果放入结果集中,最后返回这个结果集即可,对于k < 0的情况,我们是可以将其转换为k > 0的情况的,只要将数组进行翻转就可以看成是k > 0的情况,最后将计算得到的结果集也翻转一次那么就可以得到对应的答案了
3. 代码如下:
from typing import List
class Solution:
def cal(self, code: List[int], k: int):
n, res, sum = len(code), list(), 0
for i in range(1, k + 1):
sum += code[i]
for i in range(n):
res.append(sum)
t = (i + k + 1) % n
sum = sum - code[(i + 1) % n] + code[t]
return res
def decrypt(self, code: List[int], k: int) -> List[int]:
sum, n = 0, len(code)
if not k:
return [0] * n
elif k > 0:
# 当k大于0的时候
return self.cal(code, k)
else:
# 翻转数组
code = code[::-1]
# 翻转最后的结果
return self.cal(code, -k)[::-1]