80 lines
1.2 KiB
Go
80 lines
1.2 KiB
Go
|
package d4
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type cell struct {
|
||
|
|
value int
|
||
|
|
marked bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type board []cell
|
||
|
|
|
||
|
|
const SIDE_SIZE = 5
|
||
|
|
|
||
|
|
func createBoards(rawBoards []string) []board {
|
||
|
|
boards := make([]board, len(rawBoards))
|
||
|
|
|
||
|
|
for i, rawBoard := range rawBoards {
|
||
|
|
for _, cellRow := range strings.Split(rawBoard, "\n") {
|
||
|
|
for _, rawCell := range strings.Split(cellRow, " ") {
|
||
|
|
if rawCell != "" {
|
||
|
|
value, _ := strconv.Atoi(rawCell)
|
||
|
|
boards[i] = append(boards[i], cell{value, false})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return boards
|
||
|
|
}
|
||
|
|
|
||
|
|
func markBoard(b board, take int) board {
|
||
|
|
for i := range b {
|
||
|
|
if b[i].value == take {
|
||
|
|
b[i].marked = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
func gotBingo(b board) bool {
|
||
|
|
|
||
|
|
for i := 0; i < SIDE_SIZE; i++ {
|
||
|
|
if b[i*SIDE_SIZE].marked &&
|
||
|
|
b[i*SIDE_SIZE+1].marked &&
|
||
|
|
b[i*SIDE_SIZE+2].marked &&
|
||
|
|
b[i*SIDE_SIZE+3].marked &&
|
||
|
|
b[i*SIDE_SIZE+4].marked {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for j := 0; j < SIDE_SIZE; j++ {
|
||
|
|
if b[j].marked &&
|
||
|
|
b[j+SIDE_SIZE].marked &&
|
||
|
|
b[j+SIDE_SIZE*2].marked &&
|
||
|
|
b[j+SIDE_SIZE*3].marked &&
|
||
|
|
b[j+SIDE_SIZE*4].marked {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func score(b board, take int) int {
|
||
|
|
sum := 0
|
||
|
|
|
||
|
|
for _, cell := range b {
|
||
|
|
if !cell.marked {
|
||
|
|
sum += cell.value
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return sum * take
|
||
|
|
}
|