본문 바로가기
python 코딩 공부

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

by Gustav7 2024. 10. 3.

이 문제의 핵심은 마지막 원소와 마지막 전의 두 원소를 어떻게 표현할 것인가가 중점인 문제다

인덱스 문제로 num_list[-1] 인덱스를 음수로 표현하면 마지막 원소를 나타내고 -2로 나타내면 마지막 원소 전의 원소를 나타낸다.

num_list[-1] / num_list[-2]의 방법으로 마지막 부분들의 원소를 나타낼 수 있는 것을 명심하자!!!

 

def solution(num_list):
    if num_list[-1] > num_list[-2]:  # 리스트의 마지막과 마지막 전 요소 비교
        num_list.append(num_list[-1] - num_list[-2])
    else:
        num_list.append(num_list[-1] * 2)
    
    return num_list