마지막 두 원소
in Coding Test / Programmers
프로그래머스 파이썬 - 마지막 두 원소
마지막 두 원소
문제
풀이
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
파이썬은 array[-1]
이렇게 배열에 -1을 주게 되면 제일 마지막 인덱스를 가리킨다.
- -2는 마지막에서 두번째를 뜻한다.
[!note] 프로그래머스는 백준과 다르게 직접 값을 입력하는 로직을 만들지 않아도 되는것 같다.