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