map, compactMap, flatMap

안녕하세요.

 

주로 Objective-C를 쓰다 보니... Swift의 compactMap과 flatMap이 자꾸 헷갈리더라구요..

 

글로 남겨두면 안 까먹겠지 싶어 map 삼총사 공부하고 정리해 보려고요ㅎㅎ

 

---

# map

Sequence의 elements들을 변환시킨 결과의 배열을 반환하는 함수.

 

	let possibleNumbers = ["1", "2", "three", "///4///", "5"]
	let mapped: [Int?] = possibleNumbers.map { str in Int(str) }
	print(mapped) // [Optional(1), Optional(2), nil, nil, Optional(5)]

view raw mapPractice.swift hosted with ❤ by GitHub

시간복잡도는 O(n)입니다.

 

# compactMap

Sequence의 elements들을 변환시키고 nil이 아닌(=Optional이 아닌) 결과의 배열을 반환하는 함수.

(map 결괏값이랑 비교하면 어떤 차이인지 알겠죠?)

 

	let possibleNumbers = ["1", "2", "three", "///4///", "5"]
	let mapped: [Int?] = possibleNumbers.map { str in Int(str) }
	print(mapped) // [Optional(1), Optional(2), nil, nil, Optional(5)]
	let compactMapped: [Int] = possibleNumbers.compactMap { str in Int(str) }
	print(compactMapped) // [1, 2, 5]

view raw mapPractice.swift hosted with ❤ by GitHub

시간복잡도는 O(n)입니다.

 

# flatMap

Sequence의 elements들을 변환시키기고 single-level collection을 반환하는 함수.

 

여기서 single-level collection을 반환한다는 의미는, flatMap 클로저에서 sequence나 collection을 반환할 때, 이 sequence나 collection의 각 element들만 모아 하나의 배열로 반환한다는 뜻입니다.

 

	let possibleNumbers = ["1", "2", "three", "///4///", "5"]
	let mapped = possibleNumbers.map { str in [Int(str)] }
	print(mapped) // [Optional(1), Optional(2), nil, nil, Optional(5)]
	let arrayFlatMapped = possibleNumbers.flatMap { str in Array(repeating: Int(str), count: 2) }
	print(arrayFlatMapped) // [Optional(1), Optional(1), Optional(2), Optional(2), nil, nil, nil, nil, Optional(5), Optional(5)]
	let dictFlatMapped = possibleNumbers.flatMap {
	var dict = [String: String]()
	dict[$0] = $0
	return dict
	}
	print(dictFlatMapped) // [(key: "1", value: "1"), (key: "2", value: "2"), (key: "three", value: "three"), (key: "///4///", value: "///4///"), (key: "5", value: "5")]

view raw mapPractice.swift hosted with ❤ by GitHub

시간복잡도는 O(n+m)입니다.

(여기서 n은 원본 sequence length이고, m은 결괏값의 length입니다.)

 

[참고] flatMap 클로저에서 optional 값을 반환하는 경우는 deprecated 되었으니 compactMap을 사용하세요.

flatMap 클로저에서 optional 값을 반환하는 경우, compactMap과 동일한 결괏값을 반환합니다. deprecated 되었으니 compactMap을 사용하면 좋을 것 같아요ㅎㅎ

 

# 정리

API	결괏값	시간복잡도
map	elements를 변환시킨 배열	O(n) (n: 원본 sequece의 length)
compactMap	elements를 변환시킨 non-nil 배열	O(n) (n: 원본 sequece의 length)
flatMap	elements가 sequence나 collection로 변환될 경우, 내부 elements만 모은 하나의 배열을 반환	O(n+m) (n: 원본 sequece의 length, m: 결괏값의 length)

 

# 참고

map(_:) | Apple Developer Documentation

compactMap(_:) | Apple Developer Documentation

flatMap(_:) | Apple Developer Documentation

 

---

이번 글은 여기서 마무리.