aoc2021/10/code.go

package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"strings"
)

func readInput(file string) []string {
	content, err := ioutil.ReadFile(file)
	if err != nil {
		log.Fatal(err)
	}

	lines := strings.Split(string(content), "\n")
	return lines
}

var points map[rune]int

func init() {
	points = make(map[rune]int)
	points[')'] = 3
	points[']'] = 57
	points['}'] = 1197
	points['>'] = 25137
}

func parseLine(line string) rune {
	opened := make(map[rune]int)
	var lastOpened []rune

	for _, char := range line {
		if char == '(' || char == '[' || char == '{' || char == '<' {
			opened[char]++
			lastOpened = append(lastOpened, char)
			continue
		}

		if len(lastOpened) == 0 {
			return char
		}

		switch char {
		case ')':
			if lastOpened[len(lastOpened)-1] != '(' {
				return char
			}
		case ']':
			if lastOpened[len(lastOpened)-1] != '[' {
				return char
			}
		case '}':
			if lastOpened[len(lastOpened)-1] != '{' {
				return char
			}
		case '>':
			if lastOpened[len(lastOpened)-1] != '<' {
				return char
			}
		}

		lastOpened = lastOpened[:len(lastOpened)-1]
	}

	return ' '
}

func part1(input []string) (int, []string) {
	var total int
	var incomplete []string
	for _, line := range input {
		illegal := parseLine(line)
		if illegal != ' ' {
			total += points[illegal]
			continue
		}

		incomplete = append(incomplete, line)
	}

	return total, incomplete
}

func fixLine(line string) rune {
	opened := make(map[rune]int)
	var lastOpened []rune

	for _, char := range line {
		if char == '(' || char == '[' || char == '{' || char == '<' {
			opened[char]++
			lastOpened = append(lastOpened, char)
			continue
		}

		if len(lastOpened) == 0 {
			return char
		}

		switch char {
		case ')':
			if lastOpened[len(lastOpened)-1] != '(' {
				return char
			}
		case ']':
			if lastOpened[len(lastOpened)-1] != '[' {
				return char
			}
		case '}':
			if lastOpened[len(lastOpened)-1] != '{' {
				return char
			}
		case '>':
			if lastOpened[len(lastOpened)-1] != '<' {
				return char
			}
		}

		lastOpened = lastOpened[:len(lastOpened)-1]
	}

	return ' '
}

func main() {
	if len(os.Args) < 2 {
		log.Fatal("Please provide a file name as argument")
	}

	input := readInput(os.Args[1])
	total, incomplete := part1(input)
	fmt.Println("Part 1:", total)
	fmt.Println(len(incomplete))
}