Coding Test/codility
[codility] CyclicRotation c++
owls
2022. 4. 8. 15:31
728x90
- 문제 설명
크기가 A.size() 인 배열을 K 만큼 이동하라는 뜻이다.
- 문제 풀이
#include <string>
#include <vector>
#include <algorithm>
#include <iostream>
std::vector<int> solution(std::vector<int> &A, int K) {
// write your code in C++14 (g++ 6.2.0)
std::vector<int> vecResult(A.size(), 0);
for (int i = 0; i < A.size(); i++) {
vecResult[(i + K) % A.size()] = A[i];
}
return vecResult;
}
int main() {
std::vector<int> ArrayVec = { 3, 8, 9, 7, 6};
int K = 3;
std::vector<int> ArrayVec2 = {1,2,3,4};
int K2 = 4;
std::vector<int> vecResult;
vecResult = solution(ArrayVec, K);
for (const auto &it : vecResult) {
std::cout << it;
}
return 0;
}
728x90