[Swift] 프로그래머스 제일 작은 수 제거하기

문제 출처

문제 설명

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

---

풀이

배열의 가장 작은 요소를 제외하고 출력하는 간단한 문제입니다.


처음 문제를 풀었을 때 고차함수를 이용하여 간단하게 문제를 풀었습니다.

func solution(_ arr: [Int]) -> [Int] {
  return arr.count >= 2 ? arr.filter { $0 != arr.min() } : [-1]
}

filter와 min을 이용해 문제를 풀려고 했었는데 배열의 길이가 긴 케이스가 있어서 그런지 시간초과가 났습니다.
filter와 min을 중첩되게 쓴 게 문제인 거 같아 for 문을 이용해 풀었습니다…

func solution(_ arr:[Int]) -> [Int] {
  var answer: [Int] = []
  var min = arr[0]
    
  if arr.count >= 2 {
    for i in 0..<arr.count {
      if min > arr[i] {
        min = arr[i]
      }
    }
        
    for i in 0..<arr.count {
      if min != arr[i] {
        answer.append(arr[i])
      }
    }
  } else {
    return [-1]
  }
    
  return answer
}

1. 정답을 반환할 Int 형 빈 배열과 arr 배열의 첫 번째 요소를 변수로 선언합니다.
2. 배열의 길이가 2 이상일 경우 최솟값을 갱신하고 최솟값이 아닌 값들을 answer 배열에 넣습니다.
3. 배열의 길이가 1일 경우 [-1]을 반환하고 아닐 경우 answer를 반환합니다.

---

실행 결과