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
반응형