본문 바로가기
Online-Judge/Programmers

[Programmers] Python 제일 작은 수 제거하기

by nyangzzi 2022. 3. 19.
반응형

https://programmers.co.kr/learn/courses/30/lessons/12935

 

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

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

programmers.co.kr

 


 

1. 가장 작은 수를 찾는다.

min = arr[0]
    
	for i in range(len(arr)):
		if arr[i] < min:
			min = arr[i]

 

 

2. 가장 작은 수를 remove 함수를 사용하여 삭제한다.

arr.remove(min)

 

 

3. 만약 삭제한 뒤 배열이 비었다면 -1을 채운다.

    if len(arr) == 0:
        arr.append(-1)

 

 

 


 

 

[전체 코드]

def solution(arr):
    
    min = arr[0]
    
    for i in range(len(arr)):
        if arr[i] < min:
            min = arr[i]
    
    arr.remove(min)
    
    if len(arr) == 0:
        arr.append(-1)
        
    return arr
반응형

댓글