https://school.programmers.co.kr/learn/courses/30/lessons/120844
왼쪽이면 1부터 끝까지 answer에 차례대로 넣고 0번인덱스를 마지막에 넣어주고
오른쪽이면 마지막인덱스부터 넣은 뒤 0번부터 마지막 -1 을 넣어준다.
#include <string>
#include <vector>
using namespace std;
vector<int> solution(vector<int> numbers, string direction) {
vector<int> answer;
if(direction == "left")
{
for(int i = 1; i < numbers.size(); i++)
{
answer.push_back(numbers[i]);
}
answer.push_back(numbers[0]);
}
else if(direction == "right")
{
answer.push_back(numbers[numbers.size()-1]);
for(int i = 0; i < numbers.size() - 1;i++)
{
answer.push_back(numbers[i]);
}
}
return answer;
}
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> solution(vector<int> numbers, string direction) {
if(direction =="left")
{
rotate(numbers.begin(),numbers.begin()+1,numbers.end());
}
else
{
rotate(numbers.begin(),numbers.end()-1,numbers.end());
}
return numbers;
}
algorithm 헤더 내부에 있는 rotate함수를 사용하였다. +n만큼 왼쪽, -n만큼 오른쪽이다.