Set 관련 연산

Kotlin 컬렉션 패키지에는 교집합 찾기, 합치기, 차집합 구하기와 같이 set(집합)에서 자주 사용되는 연산을 위한 확장 함수가 포함되어 있습니다.

두 컬렉션을 하나로 합치려면 union() 함수를 사용하세요. 이 함수는 a union b와 같이 중위(infix) 형태로 사용할 수 있습니다. 순서가 있는 컬렉션의 경우 피연산자의 순서가 중요하다는 점에 유의하세요. 결과 컬렉션에서 첫 번째 피연산자의 요소가 두 번째 피연산자의 요소보다 앞에 위치합니다.

fun main() {
    val numbers = setOf("one", "two", "three")

    // 순서에 따른 출력
    println(numbers union setOf("four", "five"))
    // [one, two, three, four, five]
    println(setOf("four", "five") union numbers)
    // [four, five, one, two, three]
}

두 컬렉션 사이의 교집합(두 컬렉션 모두에 존재하는 요소)을 찾으려면 intersect() 함수를 사용하세요. 다른 컬렉션에 없는 요소를 찾으려면 subtract() 함수를 사용하세요. 두 함수 모두 a intersect b와 같이 중위 형태로 호출할 수 있습니다.

fun main() {
    val numbers = setOf("one", "two", "three")

    // 동일한 출력
    println(numbers intersect setOf("two", "one"))
    // [one, two]
    println(numbers subtract setOf("three", "four"))
    // [one, two]
    println(numbers subtract setOf("four", "three"))
    // [one, two]
}

두 컬렉션 중 한 곳에는 있지만 교집합에는 없는 요소(대칭 차집합으로 알려짐)를 찾으려면 union() 함수를 활용할 수도 있습니다. 이 연산을 수행하려면 두 컬렉션 간의 차집합을 각각 계산한 다음 그 결과를 합칩니다.

fun main() {
    val numbers = setOf("one", "two", "three")
    val numbers2 = setOf("three", "four")

    // 차집합 합치기 
    println((numbers - numbers2) union (numbers2 - numbers))
    // [one, two, four]
}

union(), intersect(), subtract() 함수는 리스트에도 적용할 수 있습니다. 하지만 결과는 항상 Set입니다. 결과에서 모든 중복 요소는 하나로 합쳐지며, 인덱스를 통한 접근은 불가능합니다.

fun main() {
    val list1 = listOf(1, 1, 2, 3, 5, 8, -1)
    val list2 = listOf(1, 1, 2, 2, 3, 5)

    // 두 리스트의 교집합 결과는 Set입니다
    println(list1 intersect list2)
    // [1, 2, 3, 5]

    // 동일한 요소는 하나로 합쳐집니다
    println(list1 union list2)
    // [1, 2, 3, 5, 8, -1]
}