Python/Coding Test

[Coding Test] 제일 작은 수 제거하기

gangee 2023. 4. 5. 14:55

목차

    728x90
    반응형

    문제 설명

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

    문제 풀이

    • 배열이 빈 경우, 배열에 -1을 채워 리턴해야하므로 len()을 이용해 배열의 길이를 구함
    • if 구문을 이용해 배열의 길이가 1일 경우와 아닌 경우 구하기
    • 배열의 길이가 1인 경우
    1. arr = [] 로 배열 초기화
    2. append() 통해서 -1 채워줌
    • 배열이 1이 아닌 경우
    1. min()을 이용해 배열을 최솟값 구하기
    2. remove()를 배열의 특정값 삭제하기

    정답 코드

    def solution(arr):
    
        if len(arr) == 1:
            arr=[]
            arr.append(-1)
        else:
            arr.remove(min(arr))
        return arr
    * 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.
    728x90
    반응형