THIS IS ELLIE

스위프트 Intersperse함수 본문

개발/Swift

스위프트 Intersperse함수

Ellie Kim 2021. 5. 30. 23:48

스위프트 업데이트 내용을 보다가 좋은 메서드가 있어서 정리해보려 합니다.
예전에 알고리즘 문제를 풀 때, 이런 함수 있으면 좋겠다고 생각했었는데 딱 맞는 함수가 나왔더라구요.


각 배열 사이에 - 를 넣어줘야 한다고 가정해봅시다.
저 같은 경우 루프를 돌면서 각 원소 뒤에 -를 붙여주고 if조건문을 두어 맨 마지막 원소면 -를 붙여주지 않도록 작성할 것 같네요.
또 다른 간단한 방법으로 .joined(seperator: "-")를 사용할 것 같아요.

위 예시에서 사용할 수 있는 또 다른 함수 Interperse함수를 살펴봅시다.
Intersperse는 시퀀스의 각 원소 사이에 주어진 값을 배치해줍니다.
아래와 같이 시퀀스 extension에 코드가 추가되어있습니다.

extension Sequence {
    func interspersed(with separator: Element) -> Intersperse<Self>
}

아래와 같이 interpersed(with: )를 통해 구분자를 받고 각 원소 사이에 삽입할 수 있습니다.

let numbers = [1, 2, 3].interspersed(with: 0)
// Array(numbers) == [1, 0, 2, 0, 3]

let letters = "ABCDE".interspersed(with: "-")
// String(letters) == "A-B-C-D-E"

let empty = [].interspersed(with: 0)
// Array(empty) == []

빈 문자열에 interspersed(with: )를 하면 빈 배열을 리턴해줍니다.

특히 이 메소드의 시간 복잡도는 O(1)입니다.

테스트 코드를 보려면 이쪽을 보시면 됩니다.
https://github.com/apple/swift-algorithms/blob/main/Tests/SwiftAlgorithmsTests/IntersperseTests.swift

https://github.com/apple/swift-algorithms/blob/main/Sources/Algorithms/Intersperse.swift

총 3가지의 상태를 가지며 상태에 맞게 값을 리턴해주는 것 같습니다.
내부에 어떻게 구현되어 있는지 더 자세히 확인하려면 이쪽을 보시면 됩니다.
https://github.com/apple/swift-algorithms/blob/main/Sources/Algorithms/Intersperse.swift

resource:
https://github.com/apple/swift-algorithms/blob/main/Guides/Intersperse.md

반응형

'개발 > Swift' 카테고리의 다른 글

클래스 싱글톤 vs 구조체 싱글톤  (2) 2021.12.05
스위프트 Combination  (0) 2021.07.02
스위프트 mutating 키워드  (0) 2020.11.12
스위프트 배열 크기  (0) 2020.11.09
스위프트 오버플로우 연산  (0) 2020.09.20