leetcodeSolution to some Leetcode problems written in C++ |
git clone git://git.dimitrijedobrota.com/leetcode.git |
Log | Files | Refs | README | LICENSE |
1652.cpp (521B)
0 class Solution {
1 public:
2 vector<int> decrypt(vector<int> &code, int k) {
3 const int n = size(code);
4 vector<int> res(n, 0);
6 if (k == 0) return res;
8 auto [start, end] = k >= 0 ? tuple(1, k) : tuple(n + k, n - 1);
9 int sum = 0;
11 for (int i = start; i <= end; i++)
12 sum += code[i];
13 for (int i = 0; i < n; i++) {
14 res[i] = sum;
15 sum -= code[start++ % n];
16 sum += code[++end % n];
17 }
19 return res;
20 }
21 };