[프로그래머스] Lv.0 - 마지막 두 원소.py

문제

• 정수 리스트 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합니다.

코드 1

def solution(num_list):
    result = num_list

    if int(num_list[-1]) > int(num_list[-2]):
        last_num = int(num_list[-1]) - int(num_list[-2])
        result.append(last_num)
    else:
        last_num = int(num_list[-1]) * 2
        result.append(last_num)
    answer = result
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges