code.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  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 findMaxIndex(slice []int, start, end int) int {
  25. maxIndex := start
  26. for i := start; i < end; i++ {
  27. if slice[i] > slice[maxIndex] {
  28. maxIndex = i
  29. }
  30. }
  31. return maxIndex
  32. }
  33. func part1(batteries [][]int) int {
  34. var sum int
  35. for row := range batteries {
  36. maxLeft := findMaxIndex(batteries[row], 0, len(batteries[row])-1)
  37. maxRight := findMaxIndex(batteries[row], maxLeft+1, len(batteries[row]))
  38. sum += batteries[row][maxLeft]*10 + batteries[row][maxRight]
  39. }
  40. return sum
  41. }
  42. func main() {
  43. if len(os.Args) < 2 {
  44. log.Fatal("You need to specify a file!")
  45. }
  46. filePath := os.Args[1]
  47. file, err := os.Open(filePath)
  48. if err != nil {
  49. log.Fatalf("Failed to open %s!\n", filePath)
  50. }
  51. batteries := readInput(file)
  52. fmt.Println("Part1:", part1(batteries))
  53. }