Parcourir la source

Able to read input

Piotr Czajkowski il y a 3 mois
Parent
commit
9dbca4bbe6
1 fichiers modifiés avec 52 ajouts et 0 suppressions
  1. 52 0
      02/code.go

+ 52 - 0
02/code.go

@@ -0,0 +1,52 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strconv"
+	"strings"
+)
+
+func readInput(file *os.File) [][]int {
+	scanner := bufio.NewScanner(file)
+	var reports [][]int
+
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			break
+		}
+
+		var report []int
+		numbers := strings.Split(line, " ")
+		for _, number := range numbers {
+			level, err := strconv.Atoi(number)
+			if err != nil {
+				log.Fatalf("Problem parsing input: %s", err)
+			}
+
+			report = append(report, level)
+		}
+
+		reports = append(reports, report)
+	}
+
+	return reports
+}
+
+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)
+	}
+
+	reports := readInput(file)
+	fmt.Println(reports)
+}