顺序遍历，用位运算 (n >> 31 ^ n) - (n >> 31) 求绝对值，求解《1652. 拆炸弹》
例题
1652. 拆炸弹
你有一个炸弹需要拆除，时间紧迫！你的情报员会给你一个长度为 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
答案
顺序遍历
var decrypt = function(code, k) {
const n = code.length, r = new Uint16Array(n)
if (k === 0) return r
for (let i = (k >> 31 ^ k) - (k >> 31), start = k < 0 ? n + k : 1; i--;) {
r[0] += code[start + i]
}
if (k > 0) {
for (let i = 1; i < n; i++) {
r[i] = r[i - 1] - code[i] + code[(i + k) % n]
}
} else {
for (let i = 1; i < n; i++) {
r[i] = r[i - 1] - code[(i - 1 + k + n) % n] + code[i - 1]
}
}
return r
};function decrypt(code: number[], k: number): number[] {
const n = code.length, r = new Array(n)
if (k === 0) return r.fill(0)
r[0] = 0
for (let i = (k >> 31 ^ k) - (k >> 31), start = k < 0 ? n + k : 1; i--;) {
r[0] += code[(start + i) % n]
}
if (k > 0) {
for (let i = 1; i < n; i++) {
r[i] = r[i - 1] + code[(i + k) % n] - code[i]
}
} else {
for (let i = 1; i < n; i++) {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n]
}
}
return r
};class Solution {
function decrypt($code, $k) {
$n = count($code);
$r = array_fill(0, $n, 0);
if ($k === 0) return $r;
for ($i = abs($k), $start = $k < 0 ? $n + $k : 1; $i--;) {
$r[0] += $code[($start + $i) % $n];
}
if ($k > 0) {
for ($i = 1; $i < $n; $i++) {
$r[$i] = $r[$i - 1] - $code[$i] + $code[($i + $k) % $n];
}
} else {
for ($i = 1; $i < $n; $i++) {
$r[$i] = $r[$i - 1] + $code[$i - 1] - $code[($i - 1 + $k + $n) % $n];
}
}
return $r;
}
}func decrypt(code []int, k int) []int {
n := len(code)
r := make([]int, n)
if k == 0 {
return r
}
start := 1
if k < 0 {
start = k + n
}
for i := k ^ k >> 31 - k >> 31 - 1; i >= 0; i-- {
r[0] += code[(start + i) % n]
}
if k > 0 {
for i := 1; i < n; i++ {
r[i] = r[i - 1] - code[i] + code[(i + k) % n]
}
} else {
for i := 1; i < n; i++ {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n]
}
}
return r
}class Solution {
public int[] decrypt(int[] code, int k) {
int n = code.length;
int[] r = new int[n];
if (k == 0) return r;
for (int i = Math.abs(k), start = k < 0 ? n + k : 1; i-- > 0;) {
r[0] += code[(start + i) % n];
}
if (k > 0) {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[(i + k) % n] - code[i];
}
} else {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n];
}
}
return r;
}
}public class Solution {
public int[] Decrypt(int[] code, int k) {
int n = code.Length;
int[] r = new int[n];
if (k == 0) return r;
for (int i = Math.Abs(k), start = k < 0 ? n + k : 1; i-- > 0;) {
r[0] += code[(start + i) % n];
}
if (k > 0) {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[(i + k) % n] - code[i];
}
} else {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n];
}
}
return r;
}
}int* decrypt(int* code, int codeSize, int k, int* returnSize){
int* r = malloc(sizeof(int) * codeSize);
memset(r, 0, sizeof(int) * codeSize);
*returnSize = codeSize;
if (k == 0) return r;
for (int i = abs(k), start = k < 0 ? codeSize + k : 1; i--;) {
r[0] += code[(start + i) % codeSize];
}
if (k > 0) {
for (int i = 1; i < codeSize; i++) {
r[i] = r[i - 1] + code[(i + k) % codeSize] - code[i];
}
} else {
for (int i = 1; i < codeSize; i++) {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + codeSize) % codeSize];
}
}
return r;
}class Solution {
public:
vector<int> decrypt(vector<int>& code, int k) {
int n = code.size();
vector<int> r(n, 0);
if (k == 0) return r;
for (int i = abs(k), start = k < 0 ? n + k : 1; i--;) {
r[0] += code[(start + i) % n];
}
if (k > 0) {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[(i + k) % n] - code[i];
}
} else {
for (int i = 1; i < n; i++) {
r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n];
}
}
return r;
}
};class Solution:
def decrypt(self, code: List[int], k: int) -> List[int]:
n = len(code)
r = [0] * n
if k == 0: return r
start = n + k if k < 0 else 1
for i in range(abs(k) - 1, -1, -1): r[0] += code[(start + i) % n]
if k > 0:
for i in range(1, n): r[i] = r[i - 1] + code[(i + k) % n] - code[i]
else:
for i in range(1, n): r[i] = r[i - 1] + code[i - 1] - code[(i - 1 + k + n) % n]
return r