🐍python3/Programmers

프로그래머스[코딩 기초 트레이닝]_마지막 두 원소

박필중 2023. 5. 26. 08:24
728x90

👾문제 설명

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

 

🚀제한사항

  • 2 ≤ num_list의 길이 ≤ 10
  • 1 ≤ num_list의 원소 ≤ 9

 

🌌입출력 예

num_list result
[2, 1, 6] [2, 1, 6, 5]
[5, 2, 1, 7, 5] [5, 2, 1, 7, 5, 10]

입출력 예 #1

  • 마지막 원소인 6이 그전 원소인 1보다 크기 때문에 6 - 1인 5를 추가해 return합니다.

입출력 예 #2

  • 마지막 원소인 5가 그전 원소인 7보다 크지 않기 때문에 5의 두 배인 10을 추가해 return합니다.

 

🧑‍🚀코드

def solution(num_list):
    answer = 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 answer
더보기

🤓공부

슬라이싱을 할 때 마지막 수를 가져오고 싶다면

a[-1]를 사용 혹은

a[0:-1]을 사용해 앞에서부터 마지막 수까지 가져온다.

a[:]를 사용해도 앞에서부터 마지막 수까지 가져오게 된다.

a[] 89 45 22 64 81
0 1 2 3 4
-5 -4 -3 -2 -1

 

 

728x90