From 7cd3f4232bca4165ebfddb271029115cb11bb0c4 Mon Sep 17 00:00:00 2001 From: alexchao26 Date: Sat, 3 Dec 2022 22:45:46 -0500 Subject: [PATCH] a very "good enough for AOC" kinda solution --- 2022/day02/main.go | 178 ++++++++++++++++++++++++++++++++++++++++ 2022/day02/main_test.go | 61 ++++++++++++++ 2022/day03/main.go | 95 +++++++++++++++++++++ 2022/day03/main_test.go | 64 +++++++++++++++ 4 files changed, 398 insertions(+) create mode 100644 2022/day02/main.go create mode 100644 2022/day02/main_test.go create mode 100644 2022/day03/main.go create mode 100644 2022/day03/main_test.go diff --git a/2022/day02/main.go b/2022/day02/main.go new file mode 100644 index 0000000..134e977 --- /dev/null +++ b/2022/day02/main.go @@ -0,0 +1,178 @@ +package main + +import ( + _ "embed" + "flag" + "fmt" + "strings" + + "github.com/alexchao26/advent-of-code-go/util" +) + +//go:embed input.txt +var input string + +func init() { + // do this in init (not main) so test file has same input + input = strings.TrimRight(input, "\n") + if len(input) == 0 { + panic("empty input.txt file") + } +} + +func main() { + var part int + flag.IntVar(&part, "part", 1, "part 1 or 2") + flag.Parse() + fmt.Println("Running part", part) + + if part == 1 { + ans := part1(input) + util.CopyToClipboard(fmt.Sprintf("%v", ans)) + fmt.Println("Output:", ans) + } else { + ans := part2(input) + util.CopyToClipboard(fmt.Sprintf("%v", ans)) + fmt.Println("Output:", ans) + } +} + +const ( + Win = 6 + Loss = 0 + Draw = 3 + + Rock = 1 + Paper = 2 + Scissors = 3 +) + +// this code is heinoussssss +// there's some kind of cheeky map to circular LL nodes to determine if you won or lost +// but there are so few branches to handle that it's faster to just code it manually... + +func part1(input string) int { + lines := parseInput(input) + + // opp choices: ABC rock-paper-scissors + // my choices: XYZ rock-paper-scissors + choices := map[string]int{ + "X": Rock, + "Y": Paper, + "Z": Scissors, + } + + totalScore := 0 + for _, l := range lines { + if _, ok := choices[l[1]]; !ok { + panic("choice not in choices map") + } + totalScore += choices[l[1]] + switch l[1] { + case "X": // i played rock + switch l[0] { + case "A": + totalScore += Draw + case "B": + totalScore += Loss + case "C": + totalScore += Win + default: + panic("unacceptable opp choice " + l[0]) + } + case "Y": // i played paper + switch l[0] { + case "A": // rock + totalScore += Win + case "B": // paper + totalScore += Draw + case "C": // scissors + totalScore += Loss + default: + panic("unacceptable opp choice " + l[0]) + } + case "Z": // i played scissors + switch l[0] { + case "A": // rock + totalScore += Loss + case "B": // paper + totalScore += Win + case "C": // scissors + totalScore += Draw + default: + panic("unacceptable opp choice " + l[0]) + } + } + } + + return totalScore +} + +func part2(input string) int { + /* + second column is result, not your choice + X -> you lose + Y -> draw + Z -> win + */ + lines := parseInput(input) + + winningScores := map[string]int{ + "X": Loss, + "Y": Draw, + "Z": Win, + } + + totalScore := 0 + for _, l := range lines { + if _, ok := winningScores[l[1]]; !ok { + panic("unacceptable result " + l[1]) + } + totalScore += winningScores[l[1]] + // switch on opp choice instead + switch l[0] { + case "A": // opp: rock + switch l[1] { + case "X": // lose + totalScore += Scissors + case "Y": // draw + totalScore += Rock + case "Z": // win + totalScore += Paper + default: + panic("unacceptable choice " + l[1]) + } + case "B": // opp: paper + switch l[1] { + case "X": // lose + totalScore += Rock + case "Y": // draw + totalScore += Paper + case "Z": // win + totalScore += Scissors + default: + panic("unacceptable choice " + l[1]) + } + case "C": // opp: scissors + switch l[1] { + case "X": // lose + totalScore += Paper + case "Y": // draw + totalScore += Scissors + case "Z": // win + totalScore += Rock + default: + panic("unacceptable choice " + l[1]) + } + } + } + + return totalScore +} + +func parseInput(input string) (ans [][]string) { + for _, line := range strings.Split(input, "\n") { + ans = append(ans, strings.Split(line, " ")) + } + return ans +} diff --git a/2022/day02/main_test.go b/2022/day02/main_test.go new file mode 100644 index 0000000..0d38a44 --- /dev/null +++ b/2022/day02/main_test.go @@ -0,0 +1,61 @@ +package main + +import ( + "testing" +) + +var example = `A Y +B X +C Z` + +func Test_part1(t *testing.T) { + tests := []struct { + name string + input string + want int + }{ + { + name: "example", + input: example, + want: 15, + }, + { + name: "actual", + input: input, + want: 14531, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := part1(tt.input); got != tt.want { + t.Errorf("part1() = %v, want %v", got, tt.want) + } + }) + } +} + +func Test_part2(t *testing.T) { + tests := []struct { + name string + input string + want int + }{ + { + name: "example", + input: example, + want: 12, + }, + { + name: "actual", + input: input, + want: 11258, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := part2(tt.input); got != tt.want { + t.Errorf("part2() = %v, want %v", got, tt.want) + } + }) + } +} diff --git a/2022/day03/main.go b/2022/day03/main.go new file mode 100644 index 0000000..5c73389 --- /dev/null +++ b/2022/day03/main.go @@ -0,0 +1,95 @@ +package main + +import ( + _ "embed" + "flag" + "fmt" + "strings" + + "github.com/alexchao26/advent-of-code-go/cast" + "github.com/alexchao26/advent-of-code-go/util" +) + +//go:embed input.txt +var input string + +func init() { + // do this in init (not main) so test file has same input + input = strings.TrimRight(input, "\n") + if len(input) == 0 { + panic("empty input.txt file") + } +} + +var priorities = map[string]int{} + +// lol why am i using init like this... +func init() { + // generate priorities + for i := 0; i < 26; i++ { + priorities[cast.ASCIIIntToChar('a'+i)] = i + 1 + priorities[cast.ASCIIIntToChar('A'+i)] = i + 27 + } +} + +func main() { + var part int + flag.IntVar(&part, "part", 1, "part 1 or 2") + flag.Parse() + fmt.Println("Running part", part) + + if part == 1 { + ans := part1(input) + util.CopyToClipboard(fmt.Sprintf("%v", ans)) + fmt.Println("Output:", ans) + } else { + ans := part2(input) + util.CopyToClipboard(fmt.Sprintf("%v", ans)) + fmt.Println("Output:", ans) + } +} + +func part1(input string) int { + prioritiesSum := 0 + for _, l := range strings.Split(input, "\n") { + rightHalf := l[len(l)/2:] + if len(rightHalf) != len(l)/2 { + panic("did not divide in half: " + l) + } + for i := 0; i < len(l)/2; i++ { + leftChar := l[i : i+1] + // obv unnecessary n^2 but easily fast enough... + if strings.Contains(rightHalf, leftChar) { + prioritiesSum += priorities[leftChar] + break + } + } + } + + return prioritiesSum +} + +func part2(input string) int { + prioritiesSum := 0 + sacks := strings.Split(input, "\n") + + for i := 0; i < len(sacks); i += 3 { + set2, set3 := stringToSet(sacks[i+1]), stringToSet(sacks[i+2]) + for _, char := range strings.Split(sacks[i], "") { + if set2[char] && set3[char] { + prioritiesSum += priorities[char] + break + } + } + } + + return prioritiesSum +} + +func stringToSet(s string) map[string]bool { + set := map[string]bool{} + for _, char := range strings.Split(s, "") { + set[char] = true + } + return set +} diff --git a/2022/day03/main_test.go b/2022/day03/main_test.go new file mode 100644 index 0000000..bee386c --- /dev/null +++ b/2022/day03/main_test.go @@ -0,0 +1,64 @@ +package main + +import ( + "testing" +) + +var example = `vJrwpWtwJgWrhcsFMMfFFhFp +jqHRNqRjqzjGDLGLrsFMfFZSrLrFZsSL +PmmdzqPrVvPwwTWBwg +wMqvLMZHhHMvwLHjbvcjnnSBnvTQFn +ttgJtRGJQctTZtZT +CrZsJsPPZsGzwwsLwLmpwMDw` + +func Test_part1(t *testing.T) { + tests := []struct { + name string + input string + want int + }{ + { + name: "example", + input: example, + want: 157, + }, + { + name: "actual", + input: input, + want: 7553, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := part1(tt.input); got != tt.want { + t.Errorf("part1() = %v, want %v", got, tt.want) + } + }) + } +} + +func Test_part2(t *testing.T) { + tests := []struct { + name string + input string + want int + }{ + { + name: "example", + input: example, + want: 70, + }, + { + name: "actual", + input: input, + want: 2758, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := part2(tt.input); got != tt.want { + t.Errorf("part2() = %v, want %v", got, tt.want) + } + }) + } +}