[Level1] 제일 작은 수 제거하기

문제

배열 arr에서 제일 작은 수를 제거하고 출력한다.

제거한 뒤 빈 배열이면 -1을 넣어서 출력한다.

 

입력

[4, 3, 2, 1]

[10]

 

출력

[4, 3, 2]

[-1]

---

1. arr에서 제일 작은 수를 찾는다.

2. 입력된 배열 arr을 탐색하여 제일 작은 수를 제외하고 answer에 넣는다.

3. for문 이후에 answer가 비어있다면 -1을 넣는다.

 

python

def solution(arr):
    answer = []
    
    min_num=min(arr)
    for num in arr:
        if min_num==num:
            continue
        else:
            answer.append(num)
    
    if answer==[]:
        answer.append(-1)
    return answer