day(04): simplify
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
47f865d9aa
commit
2d4eb2b19e
1 changed files with 58 additions and 101 deletions
|
@ -2,10 +2,11 @@ package day04
|
|||
|
||||
import readInput
|
||||
|
||||
fun main() {
|
||||
class Bingo(description: List<String>) {
|
||||
private val draws: List<Int> = description[0].split(",").map { it.toInt() }
|
||||
class Bingo(description: List<String>) {
|
||||
private var nextDraw: Int = 0
|
||||
private var orderOfWin: MutableList<MutableList<MutableList<Int>>> = mutableListOf()
|
||||
|
||||
private val draws: List<Int> = description[0].split(",").map { it.toInt() }
|
||||
private val boards: List<MutableList<MutableList<Int>>> = description.drop(1).chunked(6)
|
||||
.map { board ->
|
||||
board.drop(1)
|
||||
|
@ -18,9 +19,10 @@ fun main() {
|
|||
}
|
||||
.toMutableList()
|
||||
}
|
||||
private var orderOfWin: MutableList<MutableList<MutableList<Int>>> = mutableListOf()
|
||||
|
||||
fun mark(board: MutableList<MutableList<Int>>, number: Int) {
|
||||
|
||||
// TODO: Clean up
|
||||
private fun mark(board: MutableList<MutableList<Int>>, number: Int) {
|
||||
for (i in board.indices) {
|
||||
for (j in board[i].indices) {
|
||||
if (board[i][j] == number) {
|
||||
|
@ -36,90 +38,45 @@ fun main() {
|
|||
}
|
||||
}
|
||||
|
||||
fun isWinningBoard(board: MutableList<MutableList<Int>>): Boolean {
|
||||
for (i in board.indices) {
|
||||
val inRow = board[i].count { it == -1 }
|
||||
val inColumn = (board.indices).count { board[it][i] == -1 }
|
||||
|
||||
if (inRow == board.size || inColumn == board.size) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
fun winningBoardsCount(): Int {
|
||||
var counter = 0
|
||||
|
||||
for (board in boards) {
|
||||
if (isWinningBoard(board)) {
|
||||
counter++
|
||||
}
|
||||
}
|
||||
|
||||
return counter
|
||||
}
|
||||
|
||||
fun boardsCount(): Int {
|
||||
return boards.size
|
||||
private fun isWinningBoard(board: MutableList<MutableList<Int>>): Boolean = board.indices.any { i ->
|
||||
board[i].count { it == -1 } == board.size || board.indices.count { board[it][i] == -1 } == board.size
|
||||
}
|
||||
|
||||
val winningBoardsCount: Int
|
||||
get() = boards.count { isWinningBoard(it) }
|
||||
val boardsCount: Int
|
||||
get() = boards.size
|
||||
val hasWinner: Boolean
|
||||
get() {
|
||||
for (board in boards) {
|
||||
if (isWinningBoard(board)) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
get() = boards.any { isWinningBoard(it) }
|
||||
|
||||
fun draw() {
|
||||
val number = draws[nextDraw++]
|
||||
|
||||
for (board in boards) {
|
||||
mark(board, number)
|
||||
}
|
||||
boards.forEach { mark(it, number) }
|
||||
}
|
||||
|
||||
fun sumOfUnmarked(): Int {
|
||||
for (board in boards) {
|
||||
if (!isWinningBoard(board)) {
|
||||
continue
|
||||
}
|
||||
return board.sumOf { row -> row.filter { it != -1 }.sum() }
|
||||
}
|
||||
return -1
|
||||
}
|
||||
|
||||
fun sumOfLast(): Int {
|
||||
return orderOfWin.last().sumOf { row -> row.filter { it != -1 }.sum() }
|
||||
}
|
||||
|
||||
fun lastDraw(): Int {
|
||||
val lastDraw: Int
|
||||
get() {
|
||||
check(nextDraw > 0)
|
||||
return draws[nextDraw - 1]
|
||||
}
|
||||
}
|
||||
fun sumOfLast(): Int = orderOfWin.last().sumOf { row -> row.filter { it != -1 }.sum() }
|
||||
}
|
||||
|
||||
fun part1(input: Bingo): Int {
|
||||
fun part1(input: Bingo): Int {
|
||||
while (!input.hasWinner) {
|
||||
input.draw()
|
||||
}
|
||||
return input.sumOfLast() * input.lastDraw
|
||||
}
|
||||
|
||||
return input.sumOfUnmarked() * input.lastDraw()
|
||||
}
|
||||
|
||||
fun part2(input: Bingo): Int {
|
||||
while (input.winningBoardsCount() < input.boardsCount()) {
|
||||
fun part2(input: Bingo): Int {
|
||||
while (input.winningBoardsCount < input.boardsCount) {
|
||||
input.draw()
|
||||
}
|
||||
return input.sumOfLast() * input.lastDraw
|
||||
}
|
||||
|
||||
return input.sumOfLast() * input.lastDraw()
|
||||
}
|
||||
|
||||
fun main() {
|
||||
val testInput = Bingo(readInput(4, "test_input"))
|
||||
val input = Bingo(readInput(4, "input"))
|
||||
|
||||
|
|
Loading…
Reference in a new issue