[프로그래머스] 위장 (42578)(Kotlin)

원본 문제

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

종류	이름
얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

제한 사항

• clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
• 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
• 같은 이름을 가진 의상은 존재하지 않습니다.
• clothes의 모든 원소는 문자열로 이루어져 있습니다.
• 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 _ 로만 이루어져 있습니다.
• 스파이는 하루에 최소 한 개의 의상은 입습니다.

예제 입출력

numbers	return
[[yellow_hat, headgear], [blue_sunglasses, eyewear], [green_turban, headgear]]	5
[[crow_mask, face], [blue_sunglasses, face], [smoky_makeup, face]]	3

문제 풀이1 (시간초과 - 테스트케이스1)

class Solution {
    val list: MutableList<Int> = mutableListOf()

    fun solution(clothes: Array<Array<String>>): Int {
        var map: MutableMap<String, Int> = mutableMapOf()

        clothes.forEach {
            if (map.containsKey(it[1]))
                map[it[1]] = map[it[1]]!!.plus(1)
            else
                map[it[1]] = 1
        }

        for (i in map) {
            list.add(i.value)
        }

        var sum = 0
        for (i in 1 .. map.size) {
            sum += combination(i, map.size, 0)
        }

        return sum
    }

    fun combination(select: Int, depth: Int, index: Int): Int {
        if (select == 0) return 1
        if (index == depth) return 0
        if (select > 0 && index >= depth) return 0

        var sum = 0

        for (i in 0 until list[index])
            sum += combination(select - 1, depth, index + 1)

        if (select <= depth - index - 1) {
            sum += combination(select, depth, index + 1)
        }

        return sum
    }
}

문제 풀이2

class Solution {
    fun solution(clothes: Array<Array<String>>): Int {
        return clothes.groupBy { it[1] }.values.fold(1) {
            total, next -> total + (next.size + 1)
        } - 1
    }
}