[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]
풀이
배열의 가장 작은 요소를 제외하고 출력하는 간단한 문제입니다.
처음 문제를 풀었을 때 고차함수를 이용하여 간단하게 문제를 풀었습니다.
1
2
3
func solution(_ arr: [Int]) -> [Int] {
return arr.count >= 2 ? arr.filter { $0 != arr.min() } : [-1]
}
filter와 min을 이용해 문제를 풀려고 했었는데 배열의 길이가 긴 케이스가 있어서 그런지 시간초과가 났습니다.
filter와 min을 중첩되게 쓴 게 문제인 거 같아 for 문을 이용해 풀었습니다…
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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
}
- 정답을 반환할 Int 형 빈 배열과 arr 배열의 첫 번째 요소를 변수로 선언합니다.
- 배열의 길이가 2 이상일 경우 최솟값을 갱신하고 최솟값이 아닌 값들을 answer 배열에 넣습니다.
- 배열의 길이가 1일 경우 [-1]을 반환하고 아닐 경우 answer를 반환합니다.
댓글남기기