diff --git a/03/03.py b/03/03.py new file mode 100644 index 0000000..d5dfd8a --- /dev/null +++ b/03/03.py @@ -0,0 +1,21 @@ +import re + + +data = open("input.txt").read().strip() + +def mul(a, b): + return a*b + +print(sum(eval(x) for x in re.findall("mul\(\d{1,3},\d{1,3}\)", data))) + +q = "mul\(\d{1,3},\d{1,3}\)|do\(\)|don't\(\)" +f = True +s = 0 +for x in re.findall(q, data): + if x=='do()': + f = True + elif x=='don\'t()': + f = False + elif f: + s += eval(x) +print(s)