文章目錄

• • 1. 題目
  • 2. 解題

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. 解題

• 將數組拼接一次，計算前綴和、后綴和即可

class Solution { public:vector<int> decrypt(vector<int>& code, int k) {int n = code.size();if(k == 0){vector<int> ans(n, 0);return ans;}vector<int> temp(code);for(auto c : code)temp.push_back(c);//拼接數組vector<int> presum(n, 0), tailsum(n, 0);if(k < 0){k = -k;int sum = 0;for(int i = n-k; i < n; i++)sum += temp[i];//前k個的和for(int i = n; i < 2*n; i++){presum[i-n] = sum;sum -= temp[i-k];//左邊刪除sum += temp[i];//右邊新進的增加}return presum;}else{int sum = 0;for(int i = n-1+k; i >= n; i--)sum += temp[i];//后k個數的和for(int i = n-1; i >= 0; i--){tailsum[i] = sum;sum -= temp[i+k];//右邊的刪除sum += temp[i];//左邊的新進的增加}return tailsum;}} };

4 ms 8.5 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 1652. 拆炸弹（前缀和）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。