[프로그래머스/C++]Lv.0 - 배열 회전시키기

문제 링크

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만큼 오른쪽이다.