[Swift] 프로그래머스 Lv.1 : 고차함수 연습
Swift 언어로 알고리즘 문제를 풀 때 고차함수를 이용하면 간단하게 풀이할 수 있다.
내가 푼 풀이에서 1줄짜리 코드가 되는 걸 보며 고차함수에 조금 더 익숙해지고 연습해야겠다는 생각이 절실히 들었다.
오늘 풀었던 알고리즘 문제들을 고차함수로도 연습해 본 코드를 정리해 봤다.
Lv.1 : 수박수박수박수박수박수?
나의 풀이
func solution(_ n:Int) -> String {
let su = "수"
let bak = "박"
var result = ""
for i in 1...n {
result += i % 2 == 0 ? bak : su
}
return result
}
for문으로 1부터 n까지의 범위를 반복하며 짝수, 홀수를 판별하여 "수"와 "박"을 이어 붙이도록 작성했다.
다른 사람의 풀이
func solution1(_ n:Int) -> String {
return "\(String(repeating: "수박", count: n / 2))\(n % 2 == 0 ? "" : "수")"
}
func solution2(_ n:Int) -> String {
return (0..<n).map{($0%2==0 ? "수" : "박")}.reduce("",+)
}
- 1. String(repeating:count:) 메서드 사용
repeating : "수박" → "수박" 문자열 반복
count: n/2 → 반복 횟수 설정
주어진 숫자가 홀수일 때는 "수박" 패턴을 n/2번 반복하고, 짝수일 때는 "수박" 패턴을 n/2번 반복한 뒤에 "수"를 추가하는 방법이다. repeating을 사용하여 간단하게 반복하여 사용할 수 있다.
- 2. 고차함수 map, reduce 사용
.map{($0%2==0 ? "수" : "박")} → n회 반복하는 범위에서 각 요소가 짝수인지 홀수인지 판별하여 "수", "박"으로 매핑
.reduce("", +) → 생성된 배열의 모든 요소를 이어 붙여 하나의 문자열로 반환
Lv.1 : 내적
나의 풀이
// a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1] n은 a,b 배열의 길이
func solution(_ a:[Int], _ b:[Int]) -> Int {
let n = a.count
var result = 0
for i in 0..<n {
result += a[i] * b[i]
}
return result
}
반복문으로 배열의 길이만큼 요소를 반복했고,
각 배열의 같은 위치에 있는 요소를 곱하여 합산했다.
다른 사람의 풀이💡
func solution1(_ a:[Int], _ b:[Int]) -> Int {
return zip(a, b).map(*).reduce(0, +)
}
- zip 함수 : 주어진 두 배열의 각 요소를 한 쌍으로 묶어주는 역할
- map(*) : 각 시퀀스의 요소를 곱하는 역할 → (1,4)는 1*4
- reduce(0, +) : 이전까지의 합산 값을 초기값으로 설정하고 각 요소를 더하여 새로운 값 반환
따라서 map으로 생성된 배열의 모든 요소를 합산하여 반환한다.
이렇게 단순하게 정확하게 작성할 수 있다니.. 🫢
Lv.1 : 약수의 합
나의 풀이
func solution(_ n:Int) -> Int {
var result = 0
guard n > 0 else { return result }
for i in 1...n {
if n % i == 0 {
result += i
}
}
return result
}
반복문을 사용하여 n의 약수를 찾고, 찾은 약수를 더하여 결과를 반환했다.
다른 사람의 풀이
func solution(_ n:Int) -> Int {
return Array(1...n).filter{n % $0 == 0}.reduce(0, +)
}
- Array(1...n) : 1부터 n까지의 배열을 생성
- .filter { n % $0 == 0 } : 배열 중에서 나누었을 때 나머지가 0인 요소들 필터링하여 새로운 배열을 생성,$0은 배열의 각 요소이다.
- .reduce(0, +) : 필터링된 배열의 요소들을 모두 합하여 결과를 반환
Lv.1 : 나머지가 1이 되는 수 찾기
나의 풀이
func reminder(_ n:Int) -> Int {
return Array(1...n).filter{n % $0 == 1}.first! // min()
}
조건에 맞는 요소를 모두 필터링했다는 점이 아쉽긴 하지만 고차함수를 이용해서 문제에 접근해 봤다.
1부터 n까지의 숫자 중에서 n을 나누었을 때 나머지가 1인 수 중에서 가장 작은 수를 찾아 반환한다.
Lv.1 : X만큼 간격이 있는 n개의 숫자
나의 풀이
func solution(_ x:Int, _ n:Int) -> [Int64] {
// x 부터 x만큼 증가 n개
var numbers: [Int64] = []
for i in 1...n {
numbers.append(Int64(x*i))
}
return numbers
}
1부터 n까지의 수를 반복하면서 x에 i를 곱한 값을 numbers 배열에 추가하는 방법으로 풀었다.
이렇게 하면 x부터 시작하여 x만큼 증가하는 숫자가 n개 만들어진다.
다른 사람의 풀이💡
func solution(_ x:Int, _ n:Int) -> [Int64] {
return Array(1...n).map { Int64($0 * x) }
}
1부터 n까지의 숫자 배열을 생성한 후,
배열의 각 요소에 x를 곱한 값을 Int64 타입으로 변환하여 새로운 배열을 반환한다.
지금 보니 제일 편하게 생각할 수 있는 문법인 for문을 자주 사용하게 된다.
고차함수를 이용해 더 간결하고 함수형적인 스타일로 문제를 풀어보려는 습관을 가져야겠다.