소소한 컴퓨터 이야기

제일 작은 수 제거하기

by Cori

문제

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

풀이

1. Me

def solution(arr):
    arr.remove(min(arr))
    return arr or [-1]

처음에는 리스트를 내림차순 정렬하고, 리스트 슬라이싱을 이용하여 반환하였다. 그랬더니 테스트케이스에서 다 틀림.. 정렬을 그대로 유지해야 정답으로 인정되던 것.. 

 

2. Others

def rm_small(mylist):
    return [i for i in mylist if i > min(mylist)]

배열을 탐색하며 최솟값보다 큰 경우를 찾아 리스트에 담아 반환하였다. 

'CS > Coding Test' 카테고리의 다른 글

팩토리얼 진법  (0) 2021.09.25
이상한 문자 만들기  (0) 2021.09.09
스킬트리  (0) 2021.09.06
소수 만들기  (0) 2021.09.04
올바른 괄호  (0) 2021.09.04

블로그의 정보

코딩하는 오리

Cori

활동하기