문제 설명
정수를 저장한 배열, 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):
slist = sorted(arr) # 배열을 오름차순 정리
for i in arr: # 배열에서 원소 하나씩 꺼내며 정렬된 배열의 0인덱스 값(배열에서 가장 작은값)을 발견하면 삭제함
if i==slist[0]:
arr.remove(i)
if len(arr) == 0: # 작은값 삭제했는데 배열에 아무것도 없으면 [-1] 리턴
arr = [-1]
return arr
>>> 참고
sort안쓰고 min()함수 이용해서 가장 작은값 찾아도 됨
'프로그래머스' 카테고리의 다른 글
[프로그래머스][1단계][python] 평균 구하기 문제 (0) | 2023.04.05 |
---|---|
[프로그래머스][1단계][python] 콜라츠 추측 문제 (0) | 2023.04.05 |
[프로그래머스][1단계][python] 정수 제곱근 판별 문제 (0) | 2023.04.04 |
[프로그래머스][1단계][python] 정수 내림차순으로 배치하기 문제 (0) | 2023.03.30 |
[프로그래머스][1단계][python] 자연수 뒤집어 배열로 만들기 문제 (0) | 2023.03.30 |