code.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "log"
  6. "os"
  7. )
  8. func readInput(file *os.File) [][]int {
  9. scanner := bufio.NewScanner(file)
  10. var batteries [][]int
  11. for scanner.Scan() {
  12. line := scanner.Text()
  13. if line == "" {
  14. break
  15. }
  16. batteriesRow := make([]int, len(line))
  17. for i, char := range line {
  18. batteriesRow[i] = int(char - '0')
  19. }
  20. batteries = append(batteries, batteriesRow)
  21. }
  22. return batteries
  23. }
  24. func part1(batteries [][]int) int {
  25. var sum int
  26. for row := range batteries {
  27. var maxLeft int
  28. for col := 0; col < len(batteries[row])-1; col++ {
  29. if batteries[row][col] > batteries[row][maxLeft] {
  30. maxLeft = col
  31. if batteries[row][maxLeft] == 9 {
  32. break
  33. }
  34. }
  35. }
  36. maxRight := maxLeft + 1
  37. for col := maxRight; col < len(batteries[row]); col++ {
  38. if batteries[row][col] > batteries[row][maxRight] {
  39. maxRight = col
  40. if batteries[row][maxRight] == 9 {
  41. break
  42. }
  43. }
  44. }
  45. sum += batteries[row][maxLeft]*10 + batteries[row][maxRight]
  46. }
  47. return sum
  48. }
  49. func main() {
  50. if len(os.Args) < 2 {
  51. log.Fatal("You need to specify a file!")
  52. }
  53. filePath := os.Args[1]
  54. file, err := os.Open(filePath)
  55. if err != nil {
  56. log.Fatalf("Failed to open %s!\n", filePath)
  57. }
  58. batteries := readInput(file)
  59. fmt.Println("Part1:", part1(batteries))
  60. }