Python/Coding Test
[Coding Test] 제일 작은 수 제거하기
gangee
2023. 4. 5. 14:55
728x90
반응형
문제 설명
정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요. 단, 리턴하려는 배열이 빈 배열의 경우엔 배열에 -1을 채워 리턴하세요.
문제 풀이
- 배열이 빈 경우, 배열에 -1을 채워 리턴해야하므로 len()을 이용해 배열의 길이를 구함
- if 구문을 이용해 배열의 길이가 1일 경우와 아닌 경우 구하기
- 배열의 길이가 1인 경우
- arr = [] 로 배열 초기화
- append() 통해서 -1 채워줌
- 배열이 1이 아닌 경우
- min()을 이용해 배열을 최솟값 구하기
- remove()를 배열의 특정값 삭제하기
정답 코드
def solution(arr):
if len(arr) == 1:
arr=[]
arr.append(-1)
else:
arr.remove(min(arr))
return arr
* 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.
728x90
반응형