contains(_:)

생성일

2024/07/09 14:26

태그

Grammar

contains(_:)

Array나 Set 등의 Collection에서 내가 원하는 element를 찾고자 하는 경우가 있다.

스위프트에선 이때 사용하는 함수가 contains(_:) 함수이다.

Array에서의 contains(_:)

contains(Element) → Bool

func contains(_ element: Element) -> Bool
// Available when Element conforms to Equatable.
// Equatable : 서로 비교할 수 있다면 다 된다는 뜻
Swift
복사

위와 같이 사용하게 되면, Array Sequence에 찾고자 하는 element가 있는지 확인하고, Boolean 값을 리턴해준다고 한다.

let arrayInt: [Int] = [1, 2, 3, 4, 5]
let target: Int = 3

arrayInt.contains(target) ? print("\(target) 있음!") : print("\(target) 없음!")
// 3 있음!
Swift
복사

Int형 뿐만 아니라 다른 타입들도 가능하다.

let arrayDouble: [Double] = [1.0, 2.0, 3.0, 4.0]
let target: Double = 3.0

if !arrayDouble.contains(target) {
    print("\(target)있음!")
} else {
    print("\(target)없음!")
}
// 3.0있음!
Swift
복사

contains(where: (Element) → Bool) → Bool

func contains(where predicate: (Element) throws -> Bool) rethrows -> Bool
Swift
복사

주어진 predicate를 Element가 만족하는지 판별해서 Bool 타입의 값으로 리턴해준다

•

클로저를 이용해서 조건을 형성해 줄 수 있음을 의미한다.

let arrayInt: [Int] = [1, 2, 3, 4, 5]

let hadEvenNum = arrayInt.contains { element -> Bool in
		if element % 2 == 0 {
				print("true")
				return true
		}
		else {
				print("false")
				return false
		}
}
print(hadEvenNum) // 'hadEvenNum' == true
Swift
복사

위와 같이, Sequence에서 루프문을 통해 각 요소에 접근해 후행 클로저에 주어진 조건이 만족하는 경우에 메서드가 결과 값을 반환하고 종료된다.

이때, contains 메서는 O(n)의 복잡도를 가진다.

추가로 클로저를 사용하니 당연히 인자 값을 축약해서 사용할 수도 있다.

let nums = [1, 2, 3, 4, 5]
let hadOddNum = nums.contains { $0 % 2 == 1 }
// 'hadOddNum' == true
Swift
복사

Set에서의 contains(_:)

func contains(_ member: Element) -> Bool
Swift
복사

→ 복잡도는 O(1)

Set

@frozen struct Set<Element> where Element: Hashable
Swift
복사

Set이 Hashable 프로토콜을 따르기 때문이다.

Hash는 hash function을 통해 key & value 쌍을 이루어 데이터를 저장하는 자료구조

Hash를 구현할 때, Hash Collision(해시 충돌)을 방지하기 위한 Chaining이 일어나고, 최악의 경우 Hash의 LinkedList를 차례로 살펴보아야 한다면, O(n)의 복잡도를 가지게 되겠지만, 평균적으로 삽입, 삭제, 검색을 하는데 O(1) 의 복잡도를 가지게 된다.

따라서 Swift에서 제공하는 Set 또한 Hashable 프로토콜을 사용하기 때문에, Search의 기능을 수행하는 contains(_:) 를 사용하면 평균적으로 O(1) 의 복잡도를 가진다.

String에서의 contains(_:)

var string = "Swift is awesome!"
let target = "some"

if string.contains(target) {
		print("\(target) 있음!")
}
Swift
복사

Reference

[Swift] contains(_:) 메서드

Swift에서 API로 제공하는 contains(_:)메서드를 활용하자!

https://hoya-kim.github.io/2020/07/08/swift-contains/