day(10): reformat and refactor
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
91e5a765ec
commit
5a2ca59ee1
1 changed files with 21 additions and 36 deletions
|
@ -3,64 +3,49 @@ package year2021.day10
|
||||||
import readInput
|
import readInput
|
||||||
|
|
||||||
val CLOSING: Map<Char, Char> = mapOf(
|
val CLOSING: Map<Char, Char> = mapOf(
|
||||||
')' to '(',
|
')' to '(', ']' to '[', '}' to '{', '>' to '<'
|
||||||
']' to '[',
|
)
|
||||||
'}' to '{',
|
val ERROR_SCORING: Map<Char, Int> = mapOf(
|
||||||
'>' to '<'
|
')' to 3, ']' to 57, '}' to 1197, '>' to 25137
|
||||||
)
|
)
|
||||||
val ERROR_SCORING: Map<Char, Int> = mapOf(')' to 3, ']' to 57, '}' to 1197, '>' to 25137)
|
|
||||||
val AUTOCOMPLETE_SCORING: Map<Char, Long> = mapOf(
|
val AUTOCOMPLETE_SCORING: Map<Char, Long> = mapOf(
|
||||||
'(' to 1,
|
'(' to 1, '[' to 2, '{' to 3, '<' to 4
|
||||||
'[' to 2,
|
|
||||||
'{' to 3,
|
|
||||||
'<' to 4
|
|
||||||
)
|
)
|
||||||
|
|
||||||
fun firstInvalid(input: String): Char? {
|
fun check(input: String): Pair<Char?, List<Char>> {
|
||||||
val stack = mutableListOf<Char>()
|
val stack = mutableListOf<Char>()
|
||||||
|
|
||||||
for (c in input) {
|
for (c in input) {
|
||||||
if (CLOSING.contains(c)) {
|
if (CLOSING.contains(c)) {
|
||||||
// found a closing one
|
// found a closing one
|
||||||
if (stack.size < 1 || stack.last() != CLOSING[c]) {
|
if (stack.size < 1 || stack.last() != CLOSING[c]) {
|
||||||
return c
|
return Pair(c, stack)
|
||||||
}
|
}
|
||||||
stack.removeLast()
|
stack.removeLast()
|
||||||
} else {
|
} else {
|
||||||
stack.add(c)
|
stack.add(c)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return null
|
return Pair(null, stack)
|
||||||
}
|
}
|
||||||
|
|
||||||
fun part1(input: List<String>): Int = input.sumOf {
|
fun part1(input: List<String>): Int = input.sumOf {
|
||||||
firstInvalid(it)?.let { c ->
|
check(it).let { (c, _) ->
|
||||||
ERROR_SCORING[c]
|
ERROR_SCORING.getOrDefault(c, 0)
|
||||||
} ?: 0
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fun getRemaining(input: String): List<Char> {
|
fun part2(input: List<String>): Long =
|
||||||
val stack = mutableListOf<Char>()
|
input
|
||||||
|
.map { check(it) }
|
||||||
for (c in input) {
|
.filter { (c, s) -> c == null && s.isNotEmpty() }
|
||||||
if (CLOSING.contains(c)) {
|
.map { (_, stack) ->
|
||||||
stack.removeLast()
|
stack.reversed().fold(0.toLong()) { total, c ->
|
||||||
} else {
|
5 * total + AUTOCOMPLETE_SCORING[c]!!
|
||||||
stack.add(c)
|
}
|
||||||
|
}.sorted().let { scores ->
|
||||||
|
scores[scores.size / 2]
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
return stack
|
|
||||||
}
|
|
||||||
|
|
||||||
fun part2(input: List<String>): Long = input
|
|
||||||
.filter { firstInvalid(it) == null }
|
|
||||||
.map {
|
|
||||||
getRemaining(it).reversed().fold(0.toLong()) { total, c -> 5 * total + AUTOCOMPLETE_SCORING[c]!! }
|
|
||||||
}
|
|
||||||
.sorted().let { scores ->
|
|
||||||
scores[scores.size / 2]
|
|
||||||
}
|
|
||||||
|
|
||||||
fun main() {
|
fun main() {
|
||||||
val sample = readInput(10, "sample")
|
val sample = readInput(10, "sample")
|
||||||
|
|
Loading…
Reference in a new issue