[Programmers][Python][Level 1] 제일 작은 수 제거하기

프로그래머스의 Level 01 연습문제의 제일 작은 수 제거하기 문제입니다.

 

코딩테스트 연습 - 제일 작은 수 제거하기

 

문제 

정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요.

단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요.

예를들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴 하고, [10]면 [-1]을 리턴 합니다.

 

제한 조건

 

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

 

 

 

 

입출력 예

arr	return
[4,3,2,1]	[4,3,2]
[10]	[-1]

 

풀이

 

틀린 코드

문제에 arr = [4,3,2,1]에서 최솟값인 1을 제거하고 [4,3,2]를 리턴하기에 이미 내림차순 정렬이 되어 있고, 맨 뒤 항목만 빼고 리턴을 해주면 되는줄 알았습니다.

 

다른 테스트 케이스들의 arr를 sorted(arr, reverse=True)로 내림차순을 하고 리스트의 맨 뒤 항목만 뻬고 출력을 했지만, 테스트케이스들이 모두 실패했습니다...

 

문제를 고민을 해보다가 정렬을 하지 않고 그냥 최솟값을 제거하는것인가? 해서 해봤더니 성공했습니다...훜ㅋㅋㅋㅋ

Ex) arr = [5,2,1,6,4,2] 이 있으면 최솟값인 1을 제거하고 return 값으로는 [5,2,6,4,2]을 출력하면 됩니다.

def solution(arr):
        
    return sorted(arr, reverse=True)[:-1] or [-1]

 

성공 코드 1 

1. 첫 번째 조건은 arr에서 가장 작은 수를 제거해야 합니다.
   • 6행 : remove(x)는 리스트에에서 값이 x와 같은 첫 번째 요소를 제거합니다.
   • 문제의 두 번째 제한 조건이 인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다. 즉, 배열에 중복되는 값이 없기에 remove(x)가 사용이 가능합니다.
   • 중복되는 값이 있다면 index를 구해서 del(), pop()을 사용하시면 됩니다.
     • arr.pop(arr.index(min(arr))
2. 두 번째 조건은 arr가 빈 배열인 경우에는 -1을 리턴해주면 됩니다
   
   • 3행 : len(arr)의 크기가 1이면 하나 있는 숫자가 최솟값이여서 삭제해야 하기에 [-1]을 리턴해줬습니다.

 

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

 

성공 코드 2 - pop(), index() 사용

• pop(x) : pop()은 인자가 없이 사용하면 맨 마지막 요소를 꺼내고, pop(x)와 같이 인자를 주면 특정 위치에 있는 요소를 꺼냅니다. 
• index(x) : 요소의 위치, 즉 index를 반환해주는 함수입니다. 리스트에 x가 없을 때는 오류를 반환합니다.

 

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