diff --git a/1652. Defuse the Bomb b/1652. Defuse the Bomb new file mode 100644 index 0000000..340df1a --- /dev/null +++ b/1652. Defuse the Bomb @@ -0,0 +1,20 @@ +class Solution { + public: + vector decrypt(vector& code, int k) { + const int n = code.size(); + vector 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; + } +};