[Kotlin/프로그래머스] 코딩테스트 연습 > 연습문제 > 광물 캐기

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
class Solution {
fun solution(t: String, p: String): Int {
var answer: Int = 0

    /**
     * minerals를 5개씩 잘라서 저장
     * 배열별 D I S 갯수 별 내림차순 정렬
     * D I S 순으로 picks 차감하면서 계산
     */

    // 곡괭이 개수만큼 리스트로 변환
    var pickList = arrayOf<String>()
    picks.forEachIndexed { index, pick ->
        val name = when (index) {
            0 -> "diamond"
            1 -> "iron"
            else -> "stone"
        }
        repeat(pick) {
            pickList += name
        }
    }

    // 광물을 5개씩 자르고 각 개수를 저장
    var countArray: Array<IntArray> = arrayOf()
    var count = intArrayOf(0, 0, 0)
    minerals.forEachIndexed { index, mineral ->
        if (index % 5 == 0) {
            count = intArrayOf(0, 0, 0)
        }
        when (mineral) {
            "diamond" -> count[0]++
            "iron" -> count[1]++
            "stone" -> count[2]++
        }
        if (index % 5 == 4 || index == minerals.lastIndex) {
            countArray += count
        }
    }
    // 곡괭이 개수만큼만 쓸 수 있으므로 뒤에 광물은 자른다
    if (pickList.size < countArray.size) {
        countArray = countArray.slice(IntRange(0, pickList.size - 1)).toTypedArray()
    }

    // 다이아몬드, 철, 돌 개수 내림차순 정렬
    val sortedList = countArray.sortedWith(compareBy({ -it[0] }, { -it[1] }, { -it[2] }))

    // 계산
    var pickIndex = 0
    run loop@{
        sortedList.forEach { mineral ->
            val name = pickList.getOrNull(pickIndex) ?: return@loop
            val diamond = when (name) {
                "diamond" -> 1
                "iron" -> 5
                else -> 25
            }
            val iron = when (name) {
                "diamond" -> 1
                "iron" -> 1
                else -> 5
            }
            val stone = 1
            val temp = mineral[0] * diamond + mineral[1] * iron + mineral[2] * stone
            answer += temp
            pickIndex++
        }
    }
    return answer
}