拆炸弹

/*
 * 解题思路：
 * 1. 使用Java Stream进行数组处理。
 * 2. 根据k的值来决定是向前还是向后进行累加计算。
 * 3. 使用一个循环数组来处理索引越界的问题。
 */

import java.util.stream.IntStream;

public int[] decrypt(int[] code, int k) {
    int n = code.length;
    if (k == 0) {
        return new int[n]; // 全部填0
    }
    return IntStream.range(0, n).map(i -> {
        return IntStream.rangeClosed(1, Math.abs(k))
                .map(j -> k > 0 ? code[(i + j) % n] : code[(i - j + n) % n])
                .sum();
    }).toArray();
}