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

문제

마지막 두 원소 : 문제 링크

문제 분석

• 정수가 담긴 리스트 num_list가 주어질 때, 마지막 원소가 그전 원소보다 크면 마지막 원소에서 그전 원소를 뺀 값을 마지막 원소가 그전 원소보다 크지 않다면 마지막 원소를 두 배한 값을 추가하여 return
• 코드 가독성을 위해 l_size 라는 정수형 변수에 num_list의 가장 마지막 원소의 index를 의미하는 num_list.size() - 1을 저장.
• if ~ else 문을 통해 마지막 원소가 그전 원소보다 클 경우, 마지막 원소에서 그전 원소를 뺀 값을, 그렇지 않을 경우 마지막 원소를 두 배한 값을 num_list에 저장. 최종적으로 저장된 num_list 벡터를 return

풀이

#include <vector>

using namespace std;

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