day 7 part 2

1 files changed, 31 insertions, 0 deletions

diff --git a/07/2.tcl b/07/2.tcl
new file mode 100644
index 0000000..af6bd65
--- /dev/null
+++ b/07/2.tcl
@@ -0,0 +1,31 @@
+set fp [open "input"]
+set equations {}
+while {[gets $fp line] >= 0} {
+    set s1 [split $line ":"]
+    lappend equations $s1
+}
+
+proc can_match_impl {target numbers current} {
+    if {0 == [llength $numbers]} {
+        return [expr {$target == $current}]
+    }
+    set rest [lrange $numbers 1 end]
+    set first [lindex $numbers 0]
+    return [expr {[can_match_impl $target $rest [expr {$current * $first}]]
+                    || [can_match_impl $target $rest [expr {$current + $first}]]
+                    || [can_match_impl $target $rest "$current$first"]}]
+}
+
+proc can_match {target numbers} {
+    return [can_match_impl $target [lrange $numbers 1 end] [lindex $numbers 0]]
+}
+
+set sum 0
+foreach elem $equations {
+    set test_value [lindex $elem 0]
+    set numbers [lindex $elem 1]
+    if {[can_match $test_value $numbers]} {
+        set sum [expr {$sum + $test_value}]
+    }
+}
+puts $sum