알고리즘(백준, 프로그래머스)/[프로그래머스] Lv1

[프로그래머스] Lv1. 제일 작은 수 제거하기 - 파이썬(Python)

마법사 코딩공주 2023. 5. 8. 18:37
728x90
반응형

문제링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12935

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

📝 문제설명

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

🔗 제한 사항

- arr은 길이 1 이상인 배열입니다.
- 인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다.

🔗입출력예


👉 문제풀이

def solution(arr):
    if len(arr) == 1:
        return [-1]
    else:
        arr.remove(min(arr))   
        return arr

👉 다른사람 문제풀이(리스트 내포구문)

def rm_small(mylist):
    # 함수를 완성하세요
    return [i for i in mylist if i > min(mylist)]


# 아래는 테스트로 출력해 보기 위한 코드입니다.
my_list = [4, 3, 2, 1]
print("결과 {} ".format(rm_small(my_list)))
728x90
반응형