Browse Source

Able to read cards

Piotr Czajkowski 1 year ago
parent
commit
f11beac48f
1 changed files with 89 additions and 0 deletions
  1. 89 0
      04/code.go

+ 89 - 0
04/code.go

@@ -0,0 +1,89 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+)
+
+type Card struct {
+	id      int
+	winning []int
+	owned   []int
+}
+
+func readNumbers(part string) []int {
+	var numbers []int
+	numberParts := strings.Split(part, " ")
+	if len(numberParts) == 0 {
+		log.Fatalf("Can't split numbers: %s", part)
+	}
+
+	for i := range numberParts {
+		if numberParts[i] == "" {
+			continue
+		}
+
+		var number int
+		n, err := fmt.Sscanf(numberParts[i], "%d", &number)
+		if n != 1 || err != nil {
+			log.Fatalf("Can't read number: %s\n%s", numberParts[i], err)
+		}
+
+		numbers = append(numbers, number)
+	}
+
+	return numbers
+}
+
+func readInput(file *os.File) []Card {
+	scanner := bufio.NewScanner(file)
+	var cards []Card
+
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			break
+		}
+
+		var current Card
+		n, err := fmt.Sscanf(line, "Card %d:", &current.id)
+		if n != 1 || err != nil {
+			log.Fatalf("Failed to read card id: %s\n%s", line, err)
+		}
+
+		parts := strings.Split(line, ":")
+		if len(parts) != 2 {
+			log.Fatalf("Can't split card: %s", line)
+		}
+
+		numberParts := strings.Split(parts[1], "|")
+		if len(parts) != 2 {
+			log.Fatalf("Can't split tables: %s", line)
+		}
+
+		current.winning = append(current.winning, readNumbers(numberParts[0])...)
+		current.owned = append(current.owned, readNumbers(numberParts[1])...)
+		cards = append(cards, current)
+	}
+
+	return cards
+}
+
+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)
+
+	}
+
+	cards := readInput(file)
+	fmt.Println(cards)
+}