알고리즘
[algorithm] 프로그래머스 - 제일 작은 수 제거하기
hyuuny
2022. 6. 24. 00:03
📚 문제
제한 조건
- arr은 길이 1 이상인 배열입니다.
- 인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다.
입력 예시 1
[4,3,2,1]
출력 예시 1
[4,3,2]
입력 예시 2
[10]
출력 예시 2
[-1]
🧑🏻💻 풀이 과정
- 배열의 첫 번째 값을 변수(
min_val
)에 저장해두고, 반복하면서 가장 작은 수를 찾아remove
하자
def solution(arr):
min_val = arr[0]
for i in range(1, len(arr)):
if min_val > arr[i]:
min_val = arr[i]
arr.remove(min_val)
return [-1] if len(arr) < 1 else arr
# 테스트를 위한 코드입니다.
print(solution([4, 3, 2, 1]))
print(solution([10]))
python 내장 함수를 이용한 또 다른 풀이
def solution(arr):
arr.remove(min(arr))
return [-1] if len(arr) < 1 else arr
# 테스트를 위한 코드입니다.
print(solution([4, 3, 2, 1]))
print(solution([10]))