Create 1652. Defuse the Bomb

Author: Chayandas07

1652. Defuse the Bomb

@@ -0,0 +1,20 @@
+class Solution {
+ public:
+  vector<int> decrypt(vector<int>& code, int k) {
+    const int n = code.size();
+    vector<int> ans(n);
+    if (k == 0)
+      return ans;
+    int sum = 0;
+    int start = k > 0 ? 1 : n + k; 
+    int end = k > 0 ? k : n - 1;    
+    for (int i = start; i <= end; ++i)
+      sum += code[i];
+    for (int i = 0; i < n; ++i) {
+      ans[i] = sum;
+      sum -= code[start++ % n];
+      sum += code[++end % n];
+    }
+    return ans;
+  }
+};