프로그래머스 <마지막 두 원소> Python

Programmers <마지막 두 원소> 문제

문제 설명

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

제한사항
2 ≤ num_list의 길이 ≤ 10
1 ≤ num_list의 원소 ≤ 9

---

def solution(num_list):
    n = len(num_list)-1
    if num_list[n] > num_list[n-1]:
        num_list.append(num_list[n]-num_list[n-1])
    else:
        num_list.append(num_list[n]*2)
    return num_list

인덱스는 0부터 시작하는데 길이는 1부터 시작하므로 len(num_list)-1을 해줘야 인덱스를 정확히 추출 할 수 있다.

num_list의 길이가 5라면, n은 4. 따라서 num_list[n]은 num_list[4]이므로 마지막 인덱스.