[C++] 프로그래머스 마지막 두 원소

문제설명

정수 리스트 num_list가 주어질 때,
마지막 원소가 그전 원소보다 크면 마지막 원소에서 그전 원소를 뺀 값을
마지막 원소가 그전 원소보다 크지 않다면 마지막 원소를 두 배한 값을
추가하여 return하도록 solution 함수를 완성해주세요.

제한사항

• 2 ≤ num_list의 길이 ≤ 10
• 1 ≤ num_list의 원소 ≤ 9

입출력 예

num_list	result
[2, 1, 6]	[2, 1, 6, 5]
[5, 2, 1, 7, 5]	[5, 2, 1, 7, 5, 10]

풀이

마지막 원소의 숫자와 그전 원소를 비교하여 조건을 구현하는 간단한 구현문제 입니다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> num_list) {
    vector<int> answer = num_list;
    
    if(answer[answer.size() - 2] < answer[answer.size() - 1])
    {
        answer.push_back(answer[answer.size() - 1] - answer[answer.size() - 2]);
    }
    else
    {
        answer.push_back(answer[answer.size() - 1] * 2);
    }
    
    return answer;
}

결과