code.go 1.3 KB

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