[프로그래머스] Lv.0 C++ 배열 회전시키기 / Vector 값 추가와 삭제 (insert(), pop_back())

https://school.programmers.co.kr/learn/courses/30/lessons/120844

프로그래머스

 

< 문제 설명 >

정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다. 배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을 return하도록 solution 함수를 완성해주세요.

 

numbers	direction	result
[1, 2, 3]	"right"	[3, 1, 2]
[4, 455, 6, 4, -1, 45, 6]	"left"	[455, 6, 4, -1, 45, 6, 4]

 

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> numbers, string direction) {
    vector<int> answer;
    
    if(direction == "right")
    {
        int a = (numbers.size() - 1);
        numbers.insert(numbers.begin(), numbers[a]);
        numbers.pop_back();
    }
    
    if(direction == "left")
    {
        numbers.push_back(numbers[0]);
        numbers.erase(numbers.begin() + 0);
    }
    
    answer = numbers;
    
    return answer;
}

 

 

• 문제 풀이 방식 : direction이 right이면 numbers의 맨 뒤 값을 맨 앞에 넣어주고 맨 뒤에 값을 삭제해주고 direction이 left이면 numbers의 맨 앞 값을 맨 뒤에 넣어주고 맨 앞 값을 지워준다.

 

• insert() : 벡터에 인덱스 값을 활용하여 해당 인덱스 위치에 값을 넣어줌.
• insert() 형태 : 벡터명.insert(벡터명.begin(), 벡터명[인덱스 값])
• pop_back() : 해당 벡터 맨 뒤에 있는 요소의 값을 제거해줌.
• pop_back() 형태 : 벡터명.pop_back()