You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
671 B
25 lines
671 B
import re
|
|
|
|
lines = [l for l in open("input.txt", 'r').read().splitlines()]
|
|
|
|
summa, summa2, do = 0, 0, True
|
|
|
|
for l in lines:
|
|
x2 = [i.span() for i in re.finditer(r"mul\([0-9]{1,3},[0-9]{1,3}\)|do\(\)|don't\(\)", l)]
|
|
|
|
for start,end in x2:
|
|
if l[start:end] == 'do()':
|
|
do = True
|
|
elif l[start:end] == 'don\'t()':
|
|
do = False
|
|
elif do:
|
|
tup = l[start:end].split(',')
|
|
prod = int(tup[0][4:]) * int(tup[1][:-1])
|
|
summa2 += prod
|
|
summa += prod
|
|
else :
|
|
tup = l[start:end].split(',')
|
|
summa += int(tup[0][4:]) * int(tup[1][:-1])
|
|
|
|
print(summa)
|
|
print(summa2) |