AoC Day 4, Part 2 (Solved)

2023/04/code.go

@@ -18,14 +18,57 @@ func main() {
 // 4. with: true (part2), and user input
 // the return value of each run is printed to stdout
 func run(part2 bool, input string) any {
-	// when you're ready to do part 2, remove this "not implemented" block
+	gameIdRxp := regexp.MustCompile(`Card +(?P<ID>\d+):  ?(?P<Winning>.*) \|  ?(?P<Drawn>.*)`)
+
 	if part2 {
-		return "not implemented"
+		lines := strings.Split(strings.Trim(input, "\n"), "\n")
+
+		points := make([]int, len(lines))
+		for i, line := range lines {
+
+			matches := gameIdRxp.FindStringSubmatch(line)
+			if len(matches) == 0 {
+				continue
+			}
+
+			winning := strings.Split(matches[2], " ")
+			drawn := strings.Split(matches[3], " ")
+
+			points[i] = 0
+			for _, win := range winning {
+				if win == "" {
+					continue
+				}
+
+				for _, draw := range drawn {
+					if win == draw {
+						points[i] += 1
+					}
+				}
+			}
+		}
+
+		copies := make([]int, len(points))
+		for i := range points {
+			copies[i] = 1
+		}
+		for i, p := range points {
+			for j := i + 1; int(j) <= int(i)+p; j++ {
+				if j < len(copies) {
+					copies[j] += copies[i]
+				}
+			}
+		}
+
+		totalCards := int(0)
+		for _, count := range copies {
+			totalCards += count
+		}
+
+		return totalCards
 	}
 	// solve part 1 here
 
-	gameIdRxp := regexp.MustCompile(`Card +(?P<ID>\d+):  ?(?P<Winning>.*) \|  ?(?P<Drawn>.*)`)
-
 	sum := 0
 	for _, line := range strings.Split(input, "\n") {