31 lines
595 B
Python
31 lines
595 B
Python
from lib import get_data
|
|
from lib import ints
|
|
|
|
data = get_data(__file__)
|
|
|
|
|
|
def combs(acc, rest, part_2=False):
|
|
if rest == []:
|
|
return acc
|
|
|
|
nacc = []
|
|
for a in acc:
|
|
nacc.append(a + rest[0])
|
|
nacc.append(a * rest[0])
|
|
if part_2:
|
|
nacc.append(int(str(a) + str(rest[0])))
|
|
return combs(nacc, rest[1:], part_2)
|
|
|
|
|
|
t1, t2 = 0, 0
|
|
for line in data.splitlines():
|
|
xs = ints(line)
|
|
expected = xs[0]
|
|
if xs[0] in combs([xs[1]], xs[2:]):
|
|
t1 += xs[0]
|
|
if xs[0] in combs([xs[1]], xs[2:], True):
|
|
t2 += xs[0]
|
|
|
|
print(t1)
|
|
print(t2)
|