12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- package main
- import (
- "bufio"
- "fmt"
- "log"
- "os"
- "strings"
- )
- type kind byte
- const (
- op kind = iota
- val
- )
- type monkey struct {
- name string
- spec kind
- left string
- right string
- op byte
- value int
- }
- func readInput(file *os.File) map[string]monkey {
- scanner := bufio.NewScanner(file)
- monkeys := make(map[string]monkey)
- for scanner.Scan() {
- line := scanner.Text()
- if line == "" {
- continue
- }
- var current monkey
- if strings.ContainsAny(line, "+-/*") {
- current.spec = op
- n, err := fmt.Sscanf(line, "%s %s %c %s", ¤t.name, ¤t.left, ¤t.op, ¤t.right)
- if n != 4 || err != nil {
- log.Fatal("Can't parse (op): ", line, err)
- }
- } else {
- current.spec = val
- n, err := fmt.Sscanf(line, "%s %d", ¤t.name, ¤t.value)
- if n != 2 || err != nil {
- log.Fatal("Can't parse: ", line, err)
- }
- }
- current.name = strings.TrimRight(current.name, ":")
- monkeys[current.name] = current
- }
- return monkeys
- }
- func main() {
- if len(os.Args) < 2 {
- log.Fatal("You need to specify a file!")
- }
- filePath := os.Args[1]
- file, err := os.Open(filePath)
- if err != nil {
- log.Fatalf("Failed to open %s!\n", filePath)
- }
- monkeys := readInput(file)
- fmt.Println(monkeys)
- }
|