[프로그래머스/Kotlin] 2019 개발자 겨울 인턴십 - 튜플

문제

문제 출처

https://programmers.co.kr/learn/courses/30/lessons/64065 

문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(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]

 

입출력 예에 대한 설명입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

문제 예시와 같습니다.

입출력 예 #3

(111, 20)을 집합 기호를 이용해 표현하면 {{111}, {111,20}}이 되며, 이는 {{20,111},{111}}과 같습니다.

입출력 예 #4

(123)을 집합 기호를 이용해 표현하면 {{123}} 입니다.

입출력 예 #5

(3, 2, 4, 1)을 집합 기호를 이용해 표현하면 {{3},{3,2},{3,2,4},{3,2,4,1}}이 되며, 이는 {{4,2,3},{3},{2,3,4,1},{2,3}}과 같습니다.

 

풀이

두 가지 방법으로 풀어보았다.

첫 번째 방법

1. '{', '}', ','을 기준으로 문장을 나눠 문자열 배열을 만든다.
2. 각 숫자들의 개수를 세면서 값을 map에 저장한다.
3. 이 때 문자열 배열에 공백("")이 있을 수 있으니 조건문을 걸어 공백이 아닌 경우에만 처리한다.
4. 숫자가 등장한 횟수에 따라 해당하는 위치에 저장한다. (가장 많이 등장한 숫자가 0번 인덱스에, 가장 적게 등장한 숫자가 마지막 인덱스에 추가된다.)

class Solution {
    fun solution(s: String): IntArray {
        // 모든 숫자들의 배열
        val numbers = s.split('{','}',',')
        val counts = mutableMapOf<String,Int>()
        
        // 숫자가 총 몇 번 등장하는지 세기
        for (num in numbers) {
            if (num != "") {
                counts.put(num, counts.getOrElse(num, {0})+1)
            }
        }
        
        // 등장한 횟수에 따라 해당하는 위치에 저장
        // 많이 등장한 숫자일수록 앞에 저장한다.
        val answer = IntArray(counts.size)
        counts.forEach() {
            answer[counts.size - it.value] = it.key.toInt()
        }
        
        return answer
    }
}

두 번째 방법

1. 양 끝의 "{{,}}"을 제외한 문자열을 "},{"을 기준으로 분리해 집합 문자열 배열을 만든다.
2. 집합의 길이를 기준으로 오름차순으로 정렬한다.
3. 원소가 한 개인 집합부터 시작해 정답 배열에 이미 저장된 숫자가 아닌 경우 저장한다.

class Solution {
    fun solution(s: String): IntArray {
        // 집합별로 분리
        var setList = s.substring(2,s.length-2).split("},{")
        val answer = IntArray(setList.size)
        
        // 집합 길이로 정렬
        setList = setList.sortedBy() { it.length }
        
        // 원소가 한 개인 집합부터 확인
        for (i in setList.indices) {
            val numbers = setList[i]
            
            // 중복되는 원소가 없으므로 정답 배열에 존재하지 않는 숫자인 경우 저장
            numbers.split(",").forEach() {
                if (!answer.contains(it.toInt())) {
                    answer[i] = it.toInt()
                }
            }
        }
        
        return answer
    }
}

 

다른 사람의 풀이

앞서 설명한 두 번째 방법과 알고리즘은 동일하지만 구현한 방식이 달라 한 번 살펴보고자 한다.

 

1. 양 끝의 "{{,}}"을 제외한 문자열을 "},{"을 기준으로 분리해 집합 문자열 배열을 만든다.
2. asSequence()를 이용해 sequence로 변환한다. (Sequence<String>)
3. map을 이용해 각 집합을 "," 기준으로 분리하고 숫자 리스트로 변경한다. (Sequence<List<Int>>)
4. toList()를 이용해 sequence를 리스트로 변환한다. (List<List<Int>>)
5. sortedBy로 집합의 크기(숫자 리스트)의 크기순으로 오름차순 정렬한다.
6. fold()를 이용해 리스트의 값들을 하나씩 acc에 추가해 숫자 set으로 만든다. (acc.union(list)를 함으로써 acc에 존재하지 않는 숫자만 추가한다.)
7. IntArray로 변환해 return한다.

class Solution {
    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()
    }
}

 

참고

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/as-sequence.html
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.sequences/-sequence/
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/fold.html