[Python] 프로그래머스 - 제일 작은 수 제거하기
프로그래머스 - 제일 작은 수 제거하기
링크: 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이 제거가 다 되지 않습니다.
- 이런 제약 조건이 없으면 remove 함수를 이용할 때 오류가 날 수 있으므로 주의해서 사용해야합니다.
Leave a comment