[ Swift ] zip(_:_:)

 

Documentation 

• /Swift 
• /Swift Standard Library 
• /Collections 
• /zip(_:_:) 

 

 

 zip(_:_:)

func zip<Sequence1, Sequence2>(
    _ sequence1: Sequence1,
    _ sequence2: Sequence2
) -> Zip2Sequence<Sequence1, Sequence2> where Sequence1 : Sequence, Sequence2 : Sequence

두 개의 sequence로 구성한 sequence 쌍을 만드는 함수이다.

Array, String, Range처럼 Sequence 프로토콜을 채택한 타입이 매개변수로 온다. 즉, 순서가 있는 타입을 매개변수로 가진다

 

 

 

튜플 sequence 반환

zip을 사용해 두 sequence를 합치면 각 원소들이 동일한 인덱스 원소들과 짝을 지어 튜플 타입의 Zip2Sequence가 반환된다.

let strArray = ["가", "나"]
let intArray = [1, 2]

let zipped: Zip2Sequence<[String], [Int]> = zip(strArray, intArray)

 

 

for-in 구문에 zip 사용하기

zip은 두 개의 sequence 요소를 동시에 사용하고 싶을 때 유용하다. zip을 통해 각 원소의 쌍 sequence를 만들고 바로 반복(iterate)하면 된다.

만약 두 sequence의 길이가 다르면, 길이가 짧은 sequence에 맞춰 Zip2Sequnece가 생성된다.

let info = ["1반", "2반", "3반"]
let name = ["김영희", "이철수"]

for (i, n) in zip(info, name) {
    print(i, n, separator: " ")
}
// Prints
// 1반 김영희
// 2반 이철수

 

 

코딩테스트

코딩테스트 기초 문제를 풀다보면 String을 자주 만나게 되는데, Int를 사용하는 Array와 다르게 String은 개별 요소에 접근하려면 String.Index가 필요하다.  그래서 첫 번째 요소를 얻기 위해 startIndex 프로퍼티를 사용해야한다. 또 특정 위치의 요소에 접근하기 위해서 index(_:offsetBy:)을 통해 String.index를 구해야한다. 이외 String.Index를 반환하는 다양한 메서드들이 있다.

모든 요소를 한 번씩 접근하는 경우, O(n) 시간이 소요되는 index(_:offsetBy:)를 매번 사용하는 건 비효율적이다. 이럴 때는 for-in 구문이나 forEach 메서드를 사용하는 것이 좋다. 

만약 동시에 두 String의 요소를 필요로 한다면 어떻게 해야할까? 이럴 때 바로 zip 함수를 사용하면 된다.

let str1 = "가나다라"
let str2 = "1234"

var result = ""
let zipped = zip(str1, str2)
    
for (fir, sec) in zipped {
    result.append(fir)
    result.append(sec)
}
    
print(result)
// Prints 가1나2다3라4

 

프로그래머스

 

 

 

 

zip(_:_:)