From 3bd03ec7b38c671b188b3ef77d9cd5d9ec062703 Mon Sep 17 00:00:00 2001 From: alexchao26 Date: Sun, 4 Dec 2022 01:39:46 -0500 Subject: [PATCH] day4 range overlappy thing --- 2022/day04/main.go | 90 +++++++++++++++++++++++++++++++++++++++++ 2022/day04/main_test.go | 64 +++++++++++++++++++++++++++++ 2 files changed, 154 insertions(+) create mode 100644 2022/day04/main.go create mode 100644 2022/day04/main_test.go diff --git a/2022/day04/main.go b/2022/day04/main.go new file mode 100644 index 0000000..843fdba --- /dev/null +++ b/2022/day04/main.go @@ -0,0 +1,90 @@ +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") + } +} + +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 { + lines := parseInput(input) + ans := 0 + + for _, l := range lines { + if doesPair2ContainPair1(l[:2], l[2:]) || doesPair2ContainPair1(l[2:], l[:2]) { + ans++ + } + } + + return ans +} + +func doesPair2ContainPair1(pair1, pair2 []int) bool { + return pair1[0] >= pair2[0] && pair1[1] <= pair2[1] +} + +func part2(input string) int { + lines := parseInput(input) + ans := 0 + + for _, l := range lines { + if doesOverlap(l[:2], l[2:]) { + ans++ + } + } + + return ans +} + +func doesOverlap(pair1, pair2 []int) bool { + // sort + if pair1[0] > pair2[0] { + pair1, pair2 = pair2, pair1 + } + return pair1[1] >= pair2[0] +} + +func parseInput(input string) (ans [][]int) { + for _, line := range strings.Split(input, "\n") { + parts := strings.Split(line, ",") + leftParts := strings.Split(parts[0], "-") + rightParts := strings.Split(parts[1], "-") + ans = append(ans, []int{ + cast.ToInt(leftParts[0]), cast.ToInt(leftParts[1]), + cast.ToInt(rightParts[0]), cast.ToInt(rightParts[1]), + }) + } + return ans +} diff --git a/2022/day04/main_test.go b/2022/day04/main_test.go new file mode 100644 index 0000000..d3aca3d --- /dev/null +++ b/2022/day04/main_test.go @@ -0,0 +1,64 @@ +package main + +import ( + "testing" +) + +var example = `2-4,6-8 +2-3,4-5 +5-7,7-9 +2-8,3-7 +6-6,4-6 +2-6,4-8` + +func Test_part1(t *testing.T) { + tests := []struct { + name string + input string + want int + }{ + { + name: "example", + input: example, + want: 2, + }, + { + name: "actual", + input: input, + want: 526, + }, + } + 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: 4, + }, + { + name: "actual", + input: input, + want: 886, + }, + } + 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) + } + }) + } +}