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

문제

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

 

솔루션

#include <string>
#include <vector>

using namespace std;

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

 

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

프로그래머스