[Python] 마지막 두 원소

문제설명

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

 

코드

def solution(num_list):
    if num_list[-2] < num_list[-1]: 
        num_list.append(num_list[-1] - num_list[-2])
        return num_list
    else: 
        num_list.append(num_list[-1]*2)
        return num_list