力扣--拆炸弹1652-CSDN博客

本文链接：https://blog.csdn.net/m0_72084166/article/details/135364904

文章描述了一种解密算法，给定一个循环数组和整数密钥，通过替换每个数字为前后k个数字之和实现解密。提供了Java代码示例来演示这个过程。

摘要由CSDN通过智能技术生成

你有一个炸弹需要拆除，时间紧迫！你的情报员会给你一个长度为 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]。注意到数组是循环连接的。

class Solution {
    public int[] decrypt(int[] code, int k) {
        int len = code.length;
        int k1 = k;
        int num = 0;
        int[] arr = new int[len];
        if (k == 0) {
            return new int[len];
        } else if (k > 0) {
            int a = 0;
            while (k1-- > 0) {
                num += code[++a];
                if (a == len - 1) a = 0;
            }
            arr[0] = num;
            for (int i = 1; len > 1 && i < len; i++) {
                num += (code[(i + k) % len] - code[i]);
                arr[i] = num;

            }
        } else {//k<0
            int a = len - 1;
            while (k1++ < 0) {
                num += code[a--];
                if (a == 0) a = len - 1;
            }
            arr[0] = num;
            for (int i = 1; len > 1 && i < len; i++) {
                if (i + k > 0) num = num + code[i - 1] - code[i + k-1];
                else num = num + code[i - 1] - code[len - 1 +i + k];
                arr[i] = num;
            }
        }
        return arr;
    }
}