[프로그래머스] 마지막 두 원소(c++)

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

프로그래머스

---

---

답

#include <string>
#include <vector>

using namespace std;

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

변수 a,b를 만든다.

a는 num_list의 마지막 값을

b는 num_list의 마지막에서 두번째 값을

만들어준다.

​

만약 a>b이면 num_list에 마지막에a-b를 넣고

만약 아니라면 num_list 마짐가에 a*2를 넣는다.

​

여기서 push_back()은 백터 끝에 원소를 추가해준다.

---

끝