[Kotlin/프로그래머스/Lv.2] 튜플

https://programmers.co.kr/learn/courses/30/lessons/64065?language=kotlin 

코딩테스트 연습 - 튜플

문제

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

• (a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

1. 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
2. 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
3. 튜플의 원소 개수는 유한합니다.

원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현할 수 있습니다.

• {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}

예를 들어 튜플이 (2, 1, 3, 4)인 경우 이는

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}

와 같이 표현할 수 있습니다. 이때, 집합은 원소의 순서가 바뀌어도 상관없으므로

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}
• {{2, 1, 3, 4}, {2}, {2, 1, 3}, {2, 1}}
• {{1, 2, 3}, {2, 1}, {1, 2, 4, 3}, {2}}

는 모두 같은 튜플 (2, 1, 3, 4)를 나타냅니다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

[제한사항]

• s의 길이는 5 이상 1,000,000 이하입니다.
• s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
• 숫자가 0으로 시작하는 경우는 없습니다.
• s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
• s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
• return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

---

[입출력 예]

s                                                                                                         result

"{{2},{2,1},{2,1,3},{2,1,3,4}}"	[2, 1, 3, 4]
"{{1,2,3},{2,1},{1,2,4,3},{2}}"	[2, 1, 3, 4]
"{{20,111},{111}}"	[111, 20]
"{{123}}"	[123]
"{{4,2,3},{3},{2,3,4,1},{2,3}}"	[3, 2, 4, 1]

 

 

해결 과정

우선 문자열 s를 파싱하여 List<List<Int>> 타입의 이중 배열로 만드는 것 까지는 오케이. (파싱 방법은 다양)

 

이 후 이중 배열을 정렬 한 뒤 (원소의 사이즈 별로 오름차순) 정렬.

 

1. 첫번째 배열의 첫번째 원소는 튜플의 첫번째 원소가 된다. 
2. 두번째 배열에서 튜플의 원소를 제외한 나머지 원소가 튜플의 두번째 원소가 된다. 
3. n번째 배열까지 반복

나는 ,, 문제를 똑바로 읽지 않아서, 튜플 내에 중복되는 원소가 없다는 조건을 살펴보지 못함.

 

 따라서 튜플이 [2,3,1,2] 이렇게 주어질 때 이 중복되는 수 까지 어떻게 튜플 내에 순서대로 잘 집어넣을 수 있을까 고민하느라 시간을 많이 쓰게 됨. -> 그 결과 다음 배열의 합에서 이전 배열의 합을 뺀 값이 튜플의 원소가 되는 방식으로 구현함.

 

 하지만 원소중복을 고려하지 않고, 푼다면

 

1.배열끼리의 minus를 해서 구해도 되고,

2.배열끼리 union해서 구해도 되고,

3.이중배열을 만들지 않고 원소를 쭉 나열한다음에 groupBy를 써서 원소별 개수를 구해서 해당원소의 개수가 많으면 가장 먼저 나온 특성을 이용해 튜플을 완성하는 등 방법은 많았다.

 

소스 코드

 

package programmers

class SolutionTuple {

    fun solution(s: String): IntArray {
        val answer = mutableListOf<Int>()
        val tmp = s.replace("},{", " ").replace("{", "").replace("}", "").replace(",  ", "").split(" ")
        val list = tmp.map {
            it.split(",")
        }
            .asSequence()
            .map { v ->
                v.map { e -> e.toInt() }
            }.toList().sortedBy { it.size }

        println(list)

        list.fold(listOf<Int>()) { acc, v ->
            val diff = v.sum() - acc.sum()
            answer.add(diff)
			
            v
        }

        return answer.toIntArray()
    }
}


//다른 사람 풀이.
class SolutionTuple2 {
    fun solution(s: String): IntArray {
        return s.substring(2 until s.length - 2)
            .split("},{")
            .asSequence()
            .map { it.split(",").map { num -> num.toInt() } }
            .toList()
            .sortedBy { it.size }
            .fold(setOf<Int>()) { acc, list -> acc.union(list) }
            .toIntArray()
    }
}

class SolutionTuple3 {
    fun solution(s: String): IntArray = arrayListOf<Int>().apply {
        s.split("{", "}", ",")
            .groupBy { it }
            .toList()
            .sortedByDescending { it.second.count() }
            .also {
                it.takeLast(it.size - 1)
                    .forEach { add(it.first.toInt()) }
            }
    }.toIntArray()
}


fun main() {
    val solutionTuple = SolutionTuple3()
    val s = "{{2},{1,2},{2,1,3},{2,1,3,4}}"


    val tmp = arrayListOf<Int>().apply {
        s.substring(2 until s.length - 2)
            .split("},{", ",")
            .groupBy { it }
            .toList()
            .sortedByDescending { it.second.count() }
            .forEach { add(it.first.toInt()) }
    }.toIntArray()

    tmp.forEach{ println(it)}


    solutionTuple.solution(s).forEach { println(it) }
}