본문 바로가기
Engineering WIKI/Programmers

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

by wonos 2022. 4. 25.
 

코딩테스트 연습 - 제일 작은 수 제거하기

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

programmers.co.kr

내풀이 (실패)

def solution(arr):
    
    if len(arr) == 1:
        answer = [-1]
    else:
        answer = sorted(arr, reverse=True)[:-1]        
        
    return answer

내풀이 (성공)

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

다른방법 1

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

# 아래는 테스트로 출력해 보기 위한 코드입니다.
my_list = [4,3,2,1]
print("결과 {} ".format(rm_small(my_list)))