25 lines
401 B
Python
25 lines
401 B
Python
import re
|
|
from lib import get_data
|
|
from lib import ints
|
|
|
|
|
|
data = get_data(__file__)
|
|
|
|
t1 = 0
|
|
t2 = 0
|
|
enabled = True
|
|
r = re.compile(r"mul\(\d+,\d+\)|do\(\)|don't\(\)")
|
|
for m in r.findall(data):
|
|
if m == "do()":
|
|
enabled = True
|
|
elif m == "don't()":
|
|
enabled = False
|
|
else:
|
|
a, b = ints(m)
|
|
t1 += a * b
|
|
if enabled:
|
|
t2 += a * b
|
|
|
|
print(t1)
|
|
print(t2)
|