Level1. 제일 작은 수 제거하기

[jeeyeonLIM]

문제 설명

정수를 저장한 배열, 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]

[jeeyeonLIM]

ver1.

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

• remove 는 값 1개만 제거함. 따라서 solution([4,3,2,1,1]) 값이 들어올 때 1 1개만 제거한 리스트를 리턴함. (1값이 전부 제거되어야 함에도 불구하고)

ver2.

def solution(arr):
    if len(arr) <= 1: 
        return [-1]
    else : 
        arr=set(arr) # set으로 변환하고 
        arr.remove(min(arr)) # min값 제거해서 
        return sorted(list(arr), reverse=True) # 다시 list변환하고 sorting 해서 보여주기

• set으로 변환해서 중복값 제거한 후 min값 제거, 그 후 다시 list 변환 후 sorting 했다.
• 결과는 테케 한개도 통과 못하는 참혹함.
• 변환 과정 때문인가?

ver3.

def solution(arr):
    if len(arr) <= 1: 
        return [-1]
    else : 
        return [i for i in arr if i > min(arr)] # 한번에 반환

• 구글링으로 간단한 힌트를 얻은 후 수정
• 저렇게 간단한 방법이 있었는데 생각이 안떠올랐다.
• 시간초과는 왜 뜨는 것인지 잘 몰랐다. 더 줄일 곳이 없는거 같은데

[jeeyeonLIM]

최종 ver.

• 질문하기에서 ver3 문제랑 같은 문제를 찾았다 ❗

• list 내에서 요소 하나하나 검사할 때마다 min값을 계산해서 비교해서 연산속도가 O(n)으로 증가한다는 의미인거 같다.

• 그래서 미리 min값을 저장해놓고 그 값만 가져와서 비교하는 방식을 취해야 한다!!

  def solution(arr):
  arr_min = min(arr)
  if len(arr) <= 1: 
      answer = [-1]
  else :
      answer = [i for i in arr if i > arr_min] 
  return answer

• 드디어 통과 👍👍

• 무조건 짧은코드가 좋은 코드라고 생각했는데, 시행착오 겪어가는 과정이 의미있다.

• [i for i in arr if i > arr_min]

  • list comprehension ❗