aoc2022/05/code.go

package main

import (
	"bufio"
	"fmt"
	"log"
	"os"
)

type move struct {
	quantity int
	from     int
	to       int
}

func parseBoxes(line string, boxes [][]byte) [][]byte {
	counter := 0
	index := 0
	readLetter := false

	for i := range line {
		if line[i] == '\n' {
			break
		}

		counter++
		if counter > 3 {
			index++
			counter = 0
		}

		if line[i] == '[' {
			readLetter = true
			continue
		}

		if line[i] == ']' {
			readLetter = false
			continue
		}

		if readLetter {
			edge := len(boxes) - 1
			if index > edge {
				for i := 0; i < index-edge; i++ {
					boxes = append(boxes, []byte{})
				}
			}

			boxes[index] = append(boxes[index], line[i])
		}
	}

	return boxes
}

func reverseBoxes(boxes [][]byte) [][]byte {
	for x := range boxes {
		for i, j := 0, len(boxes[x])-1; i < j; i, j = i+1, j-1 {
			boxes[x][i], boxes[x][j] = boxes[x][j], boxes[x][i]
		}
	}

	return boxes
}

func parseMove(line string) move {
	var current move
	n, err := fmt.Sscanf(line, "move %d from %d to %d", &current.quantity, &current.from, &current.to)
	if n != 3 || err != nil {
		log.Fatal("Can't parse move!", err)
	}

	current.from--
	current.to--

	return current
}

func readInput(file *os.File) ([][]byte, []move) {
	scanner := bufio.NewScanner(file)
	var moves []move
	var boxes [][]byte
	readBoxes := true

	for scanner.Scan() {
		line := scanner.Text()
		if line == "" {
			readBoxes = false
			continue
		}

		if readBoxes {
			boxes = parseBoxes(line, boxes)
			continue
		}

		moves = append(moves, parseMove(line))
	}

	return reverseBoxes(boxes), moves
}

func moveBox(boxes [][]byte, action move) [][]byte {
	last := len(boxes[action.from]) - 1
	letter := boxes[action.from][last]
	boxes[action.from] = boxes[action.from][:last]
	boxes[action.to] = append(boxes[action.to], letter)

	return boxes
}

func part1(boxes [][]byte, moves []move) string {
	for i := range moves {
		for j := 0; j < moves[i].quantity; j++ {
			moveBox(boxes, moves[i])
		}
	}

	var word []byte
	for i := range boxes {
		word = append(word, boxes[i][len(boxes[i])-1])
	}

	return string(word)
}

func moveBoxes(boxes [][]byte, action move) [][]byte {
	toTake := len(boxes[action.from]) - action.quantity
	boxes[action.to] = append(boxes[action.to], boxes[action.from][toTake:]...)
	boxes[action.from] = boxes[action.from][:toTake]

	return boxes
}

func part2(boxes [][]byte, moves []move) string {
	for i := range moves {
		moveBoxes(boxes, moves[i])
	}

	var word []byte
	for i := range boxes {
		word = append(word, boxes[i][len(boxes[i])-1])
	}

	return string(word)
}

func copyBoxes(boxes [][]byte) [][]byte {
	var copied [][]byte

	for i := range boxes {
		copied = append(copied, make([]byte, len(boxes[i])))
		copy(copied[i], boxes[i])
	}

	return copied
}

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)

	}

	boxes, moves := readInput(file)
	originalBoxes := copyBoxes(boxes)

	fmt.Println("Part1:", part1(boxes, moves))
	fmt.Println("Part2:", part2(originalBoxes, moves))
}