[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
class Solution {
fun solution(want: Array<String>, number: IntArray, discount: Array<String>): Int {
var answer: Int = 0
val wantMap = hashMapOf<String, Int>().apply {
this.putAll(want.mapIndexed { index, s -> s to number[index] })
}
val discountMap = hashMapOf<String, Int>()
var index = 0
var isAvailable = true
while (index + 9 < discount.size) {
if (index == 0) {
for (i in 0 until 10) {
discountMap[discount[i]] = (discountMap[discount[i]] ?: 0) + 1
}
} else {
val prevKey = discount[index - 1]
discountMap[prevKey]?.let {
discountMap[prevKey] = it - 1
}
discountMap[discount[index+9]] = (discountMap[discount[index+9]] ?: 0) + 1
}
run loop@{
wantMap.forEach { (key, value) ->
if ((discountMap[key] ?: 0) < value) {
isAvailable = false
return@loop
}
}
}
if (isAvailable) {
answer++
}
index++
isAvailable = true
}

        return answer
    }
}