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

마지막 두 원소

---

문제설명

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

---

제한사항

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

---

입출력 예

입출력 예 설명

입출력 예 #1

• 마지막 원소인 6이 그전 원소인 1보다 크기 때문에 6 - 1인 5를 추가해 return합니다.

입출력 예 #2

• 마지막 원소인 5가 그전 원소인 7보다 크지 않기 때문에 5의 두 배인 10을 추가해 return합니다.

---

작성코드

#include <string>
#include <vector>

using namespace std;

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

---

문제 URL

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

프로그래머스

---

반응형