03_corruption.py 302 B

1234567891011
  1. #!/usr/bin/env python3
  2. import sys
  3. import re
  4. def mul(program):
  5. return sum(int(m[1]) * int(m[2]) for m in
  6. re.finditer(r'mul\((\d{1,3}),(\d{1,3})\)', program))
  7. program = sys.stdin.read()
  8. print(mul(program))
  9. print(mul(re.sub(r'don\'t\(\).*?(do\(\)|$)', '', program, flags=re.DOTALL)))