code.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "log"
  6. "os"
  7. )
  8. func readInput(file *os.File) ([][]int, []int) {
  9. scanner := bufio.NewScanner(file)
  10. var ranges [][]int
  11. var ingredients []int
  12. readingRanges := true
  13. for scanner.Scan() {
  14. line := scanner.Text()
  15. if line == "" {
  16. readingRanges = false
  17. continue
  18. }
  19. if readingRanges {
  20. var start, end int
  21. n, err := fmt.Sscanf(line, "%d-%d", &start, &end)
  22. if n != 2 || err != nil {
  23. log.Fatalf("Bad input: %s", line)
  24. }
  25. ranges = append(ranges, []int{start, end})
  26. } else {
  27. var ingredient int
  28. n, err := fmt.Sscanf(line, "%d", &ingredient)
  29. if n != 1 || err != nil {
  30. log.Fatalf("Bad input: %s", line)
  31. }
  32. ingredients = append(ingredients, ingredient)
  33. }
  34. }
  35. return ranges, ingredients
  36. }
  37. func main() {
  38. if len(os.Args) < 2 {
  39. log.Fatal("You need to specify a file!")
  40. }
  41. filePath := os.Args[1]
  42. file, err := os.Open(filePath)
  43. if err != nil {
  44. log.Fatalf("Failed to open %s!\n", filePath)
  45. }
  46. ranges, ingredients := readInput(file)
  47. fmt.Println(ranges, ingredients)
  48. }