[프로그래머스]Lv.1 - 제일 작은 수 제거하기_파이썬

 

제일 작은 수 제거하기

• 정수를 저장한 배열, arr에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해야 한다.

• step 1. arr의 element가 1이라면, -1을 return 한다

• step 2. 제일 작은 수를 찾는다.

• step 3. 제일 작은 수를 arr에서 뺀다. (이때, arr의 빼는 element를 제외한 나머지 index 변하면 안 된다)

---

• 제일 작은 수를 찾기 위해서 정렬 후 맨 앞에 있는 값(제일 작은 값)을 arr_min에 저장하고
• arr에서 그 element를 찾아서 빼도록 했습니다!

def solution(arr):
    answer = []
    if len(arr) == 1:
        answer.append(-1)
    
    else: 
        arr_min = sorted(arr).pop(0)
        arr.remove(arr_min)
        answer = arr

    return answer

[test code]
print(solution([7]))
print(solution([4,3,2,1]))
print(solution([-4,3,-6.4,7]))
print(solution([3,1,5,0,2]))


[out]

[-1]
[4, 3, 2]
[-4, 3, 7]
[3, 1, 5, 2]