blob: 61852d0c6a07c77c7f8c7f42f2d17b7578b1078b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import re
with open('3.txt', 'r') as f:
data = f.read()
# Regular expressions for matching patterns
mul_pattern = r'mul\(\d+,\d+\)'
do_pattern = r'do\(\)'
dont_pattern = r"don't\(\)"
multiply = True
muls = 0
en_muls = []
# Find all mul commands and do/don't instructions
matches = re.finditer(mul_pattern + '|' + do_pattern + '|' + dont_pattern, data)
for match in matches:
pattern = match.group()
print(pattern)
#identify "do()" and "dont()"
if pattern == "do()":
multiply = True
elif pattern == "don't()":
multiply = False
elif pattern.startswith('mul'):
if multiply:
en_muls.append(pattern)
print(en_muls)
ints = [l.split("mul(")[1].rstrip(')') for l in en_muls]
print(ints)
for i in ints:
n1, n2 = map(int, i.split(','))
print(n1, n2)
muls += n1 * n2
print(muls)
|
