Create 1652.cpp

This commit is contained in:
Kirigaya Kazuto 2022-09-24 14:13:47 +08:00 committed by GitHub
parent c88ede9274
commit 1be36799dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

37
LeetCode/1652.cpp Normal file
View File

@ -0,0 +1,37 @@
class Solution {
public:
vector<int> decrypt(vector<int>& code, int k) {
if (k == 0) {
return vector<int>(code.size(), 0);
}
int n = code.size();
vector<int> result;
if (k > 0) {
for (int i = 0; i < n; i++) {
int sum = 0;
for (int j = 0; j < k; j++) {
int index = (i + 1 + j) % n;
sum += code[index];
}
result.push_back(sum);
}
}
else {
for (int i = 0; i < n; i++) {
int sum = 0;
for (int j = 0; j < -k; j++) {
int index = (i - 1 - j);
if (index < 0) index += n;
sum += code[index];
}
result.push_back(sum);
}
}
return result;
}
};