aoc2022/09/code.go

package main

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

type move struct {
	direction byte
	steps     int
}

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

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

		var current move
		n, err := fmt.Sscanf(line, "%c %d", &current.direction, &current.steps)
		if n != 2 || err != nil {
			log.Fatal("Can't parse cd:", err)
		}

		moves = append(moves, current)
	}

	return moves
}

type point struct {
	x int
	y int
}

func tailInVicinity(leader *point, tail point) bool {
	if leader == nil {
		return false
	}

	return tail.x >= leader.x-1 && tail.x <= leader.x+1 && tail.y >= leader.y-1 && tail.y <= leader.y+1
}

func catchUp(leader *point, tail point) point {
	if tail.x > leader.x {
		tail.x--
	} else if tail.x < leader.x {
		tail.x++
	}

	if tail.y > leader.y {
		tail.y--
	} else if tail.y < leader.y {
		tail.y++
	}

	return tail
}

func moveRight(leader *point, tail point) point {
	if !tailInVicinity(leader, tail) {
		if leader != nil {
			tail = catchUp(leader, tail)
		} else {
			tail.x++
		}
	}

	return tail
}

func moveLeft(leader *point, tail point) point {
	if !tailInVicinity(leader, tail) {
		if leader != nil {
			tail = catchUp(leader, tail)
		} else {
			tail.x--
		}
	}

	return tail
}

func moveUp(leader *point, tail point) point {
	if !tailInVicinity(leader, tail) {
		if leader != nil {
			tail = catchUp(leader, tail)
		} else {
			tail.y++
		}
	}

	return tail
}

func moveDown(leader *point, tail point) point {
	if !tailInVicinity(leader, tail) {
		if leader != nil {
			tail = catchUp(leader, tail)
		} else {
			tail.y--
		}
	}

	return tail
}

func drawTail(leader *point, tail point, action move) point {
	switch action.direction {
	case 'R':
		return moveRight(leader, tail)
	case 'L':
		return moveLeft(leader, tail)
	case 'U':
		return moveUp(leader, tail)
	case 'D':
		return moveDown(leader, tail)
	}

	return tail
}

func wayOfTail(moves []move, snake []point) int {
	trail := make(map[point]bool)
	last := len(snake) - 1

	for i := range moves {
		for s := 0; s < moves[i].steps; s++ {
			for j := range snake {
				if j == 0 {
					snake[j] = drawTail(nil, snake[j], moves[i])
				} else {
					snake[j] = drawTail(&snake[j-1], snake[j], moves[i])
				}

			}

			trail[snake[last]] = true
		}
	}

	return len(trail)
}

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)

	}

	moves := readInput(file)
	snake1 := make([]point, 2)
	fmt.Println("Part1:", wayOfTail(moves, snake1))

	snake2 := make([]point, 10)
	fmt.Println("Part2:", wayOfTail(moves, snake2))
}