瀏覽代碼

Able to read input

Piotr Czajkowski 2 年之前
父節點
當前提交
24b00c28c9
共有 1 個文件被更改,包括 51 次插入0 次删除
  1. 51 0
      09/code.go

+ 51 - 0
09/code.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+)
+
+type move struct {
+	direction byte
+	steps     int
+}
+
+func readInput(file *os.File) []move {
+	scanner := bufio.NewScanner(file)
+	var moves []move
+
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			continue
+		}
+
+		var current move
+		n, err := fmt.Sscanf(line, "%c %d", &current.direction, &current.steps)
+		if n != 2 || err != nil {
+			log.Fatal("Can't parse cd:", err)
+		}
+
+		moves = append(moves, current)
+	}
+
+	return moves
+}
+
+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)
+
+	}
+
+	moves := readInput(file)
+	fmt.Println(moves)
+}