[Python] 프로그래머스 - 제일 작은 수 제거하기

less than 1 minute read

프로그래머스 - 제일 작은 수 제거하기


링크: https://programmers.co.kr/learn/courses/30/lessons/12935

문제 설명


정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요. 단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요. 예를들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴 하고, [10]면 [-1]을 리턴 합니다.

제한 사항


  • arr은 길이 1 이상인 배열입니다.
  • 인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다.

입출력 예제


  • arr return
    [4,3,2,1] [4,3,2]
    [10] [-1]

코드


def solution(arr):
    if len(arr) > 1:
        arr.remove(min(arr))
        return arr
    else :
        return [-1]

코드 설명


  • len(arr) > 1인 이유?
    • 배열의 길이가 1이면 가장 작은 수를 제거했을 때 리턴하려는 배열이 빈 배열이므로 1보다 큰 경우로 계산합니다.
  • 인덱스 i 와 j 가 같지 않으면 arr[i] 와 arr[j]도 같지 않기 때문에 원소 하나만 지워주는 remove() 함수를 사용할 수 있습니다.
    • 이런 제약 조건이 없으면 remove 함수를 이용할 때 오류가 날 수 있으므로 주의해서 사용해야합니다.
      • 예) arr = [4,3,2,1,1] -> remove(min(arr)) -> arr=[4,3,2,1] 이 되므로 가장 작은 수인 1이 제거가 다 되지 않습니다.

Leave a comment