blob: 9c96af9bd5b075f639f2e02ecc1dae8a17ce71b7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#!/bin/env tclsh
source lib.tcl
setup 7
puts {Part 1: Total of valid test lines}
proc op_search {ops targ cur rest} {
# cursed, produces exprs of the form "$cur<op>$next == $targ || ..."
set e1 {}
foreach op $ops {
lappend e1 [string cat {$cur} $op {$next == $targ}]
}
set e1 [join $e1 " || "]
# cursed, produces exprs of the form "[_op_search $e1 $e2 $targ [expr {$cur<op>$next}] $rest] || ..."
set e2 {}
foreach op $ops {
lappend e2 [string cat \
{[_op_search $e1 $e2 $targ [expr } \
[list [string cat {$cur} $op {$next}]] \
{] $rest]}]
}
set e2 [join $e2 " || "]
_op_search $e1 $e2 $targ $cur $rest
}
proc _op_search {e1 e2 targ cur rest} {
set rest [lassign $rest next]
if {$rest == {}} {
expr $e1
} else {
expr $e2
}
}
set test_total 0
set cat_test_total 0
foreach line $input {
lassign [split $line :] targ nums
set nums [lassign $nums first]
if [op_search {{ + } { * }} $targ $first $nums] {
incr test_total $targ
}
if [op_search {{ + } { * } {}} $targ $first $nums] {
incr test_total $targ
}
}
puts "Total: $test_total"
puts ""
puts {Part 2: Total of valid test lines with || operator}
puts "2: $"
|
