[프로그래머스] [kotlin] 위장

//fun main() {
//    println(
//        solution(
//            arrayOf(
//                arrayOf("yellowhat", "face"),
//                arrayOf("bluesunglasses", "face"),
//                arrayOf("green_turban", "face")
//            )
//        )
//    )
//}

fun solution(clothes: Array<Array<String>>): Int {
    var answer = 1
    val mmap = mutableMapOf<String, Int>()

    clothes.forEach {
        mmap[it[1]] = mmap.getOrDefault(it[1], 0) + 1
    }

    mmap.keys.forEach {
        answer *= (mmap[it]!! + 1)
    }

    return answer - 1
}

 

fun solution(clothes: Array<Array<String>>): Int {
    val hashMap = mutableMapOf<String, MutableList<String>>()
    clothes.forEach {
        if (hashMap[it[1]] == null) hashMap[it[1]] = mutableListOf()
        hashMap[it[1]]?.add(it[0])
    }
    var answer = 1
    hashMap.forEach {
        answer *= (it.value.size + 1)
    }
    return answer - 1
}