| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- package main
- import (
- "bufio"
- "fmt"
- "log"
- "os"
- "strconv"
- )
- func readInput(file *os.File) [][]byte {
- scanner := bufio.NewScanner(file)
- var batteries [][]byte
- for scanner.Scan() {
- line := scanner.Text()
- if line == "" {
- break
- }
- batteriesRow := make([]byte, len(line))
- for i := range line {
- batteriesRow[i] = line[i]
- }
- batteries = append(batteries, batteriesRow)
- }
- return batteries
- }
- func findMaxIndex(slice []byte, start, end int) int {
- maxIndex := start
- for i := start; i < end; i++ {
- if slice[i] > slice[maxIndex] {
- maxIndex = i
- }
- }
- return maxIndex
- }
- func part1(batteries [][]byte) int {
- var sum int
- for row := range batteries {
- maxLeft := findMaxIndex(batteries[row], 0, len(batteries[row])-1)
- maxRight := findMaxIndex(batteries[row], maxLeft+1, len(batteries[row]))
- num, err := strconv.Atoi(string(batteries[row][maxLeft]) + string(batteries[row][maxRight]))
- if err != nil {
- log.Fatalf("Failed to convert %s to int!\n", string(batteries[row][maxLeft])+string(batteries[row][maxRight]))
- }
- sum += num
- }
- return sum
- }
- 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)
- }
- batteries := readInput(file)
- fmt.Println("Part1:", part1(batteries))
- }
|