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

문제 설명

정수를 저장한 배열, 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]

풀이

이 문제는 제일 먼저 arr에서 제일 작은 수를 제거해야한다.

• min() 함수로 arr에서 제일 작은 수를 찾는다.
• index() 메소드로 arr에서 제일 작은 수의 인덱스를 찾는다.
• pop() 메소드로 arr에서 찾은 인덱스의 요소를 제거한다.(pop을 하게되면 그 위치 요소 삭제되고 요소가 뒤의 요소들이 한 칸씩 땡겨짐.)

이렇게 제일 작은 수를 제거하고 arr의 길이를 체크해 조건에 따라 다른 값을 반환하면 된다.

• 0일 경우 : [-1]을 반환 한다.
• 0이 아닐경우 : arr를 반환한다.

 

def solution(arr):
    arr.pop(arr.index(min(arr)))
    return [-1] if 0 == len(arr) else arr