[Swift] 고차함수 - map, filter, reduce

고차함수?

하나 이상의 함수(클로저)를 인자 로 받거나, 함수(클로저)를 반환 하는 함수로

반복문을 사용하여 구현을 할 수 있지만 상수를 사용하면 되어서 메모리 자원측에서 더 효율적으로 사용할 수 있다.

또한, 반복문으로 구현하게 되면 가독성이 떨어지고 복잡해지기 때문에, 가독성을 올리고 편리성을 높이기 위해서 사용.

 

그 중, 배열 다룰때 가장 중요한 세 가지 map, filter, reduce에 대해 알아보자

---

🔥 map

: 기존 데이터 변형하여 새로운 컨테이너를 만들 때 사용

컬렉션 타입을 순회(for문 사용)할 필요 없이 간단하게 구현 가능!

// Declaration
func map<U>(transform: (T) -> U) -> Array<U>

 

[간단 사용예제] 배열 원소 데이터 형변환하기

var string = ["1","2","3","4"]
string.map { Int($0)! } // [1,2,3,4] 각 원소를 전부 Int로 맵핑

 

 

[문제] - 배열자르기

정수 배열 numbers와 정수 num1, num2가 매개변수로 주어질 때, numbers의 num1번 째 인덱스부터 num2번째 인덱스까지 자른 정수 배열을 return 하도록 solution 함수를 완성해보세요.

 

[for문 사용]

func solution(_ numbers:[Int], _ num1:Int, _ num2:Int) -> [Int] {
	var newArr: [Int] = []
	for i in (num1...num2){
		newArr.append(numbers[i])
	}
	return newArr
}

[map 메소드 사용]

func solution(_ numbers:[Int], _ num1:Int, _ num2:Int) -> [Int] {
    return (num1...num2).map{numbers[$0]}
}

클로저 축약 문법을 사용해서 훨~씬 간결하게 코드 구현이 가능

---

🔥 filter

: 기존 값들 중 조건에 만족하는 값을 추출하여 새로운 컨테이너로 반환, 요소가 반환된 배열에 포함되어야 하는지 여부를 Bool 값으로 반환하는 클로저

// Declaration
func filter(includeElement: (T) -> Bool) -> Array<T>

 

[간단 사용예제]

var array = [1,2,3,4]
array.filter { $0 % 2 == 0 } // [2,4] 조건에 맞는 수만 뽑아냄

 

 

[문제] - 특정 문자 제거하기

문자열 my_string과 문자 letter이 매개변수로 주어집니다. my_string에서 letter를 제거한 문자열을 return하도록 solution 함수를 완성해주세요.

 

[for문 사용]

func solution(_ my_string:String, _ letter:String) -> String {
	var my_array = Array(my_string)
	var new_array: [Character] = []
        
	for char in my_array{
		if char != Character(letter){
			new_array.append(char)
		}
	}
	return String(new_array)
}

[filter 사용]

func solution(_ my_string:String, _ letter:String) -> String {
	return my_string.filter{String($0) != letter}
}

---

🔥 reduce

: 기존 컨테이너에서 내부의 값들을 결합하여 새로운 값을 만든다.

배열의 각 항목들을 재귀적으로 클로저를 적용시켜 하나의 값을 만듬.

func reduce<U>(initial: U, combine: (U, T) -> U) -> U

[간단 사용예제]

var array = [1,2,3,4]
array.reduce(0, +) // 숫자 합이 나타남. 문자열 합치기도 가능

 

[문제] - 배열 원소들의 합 구하기

[for문 사용]

let numberArray = [1,2,3,4,5,6,7,8,9,10]
var sum = 0
for number in numberArray {
    sum += number
}
print(sum)

[filter 사용]

let array = [5,6,7,8]
array.reduce(0, +)

//다른 연산도 가능
array.reduce(0, -)

// (1) * (5*6*7*8)
array.reduce(1, *)

// (16800) / (5 /6 /7 /8) = 10
array.reduce(16800, /))