blob: 5eea1319e558dc41cb319273dd2d5c5d351a0715 (
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
42
43
44
45
46
47
48
|
#!/usr/bin/env python3
from functools import reduce
from re import findall
from copy import deepcopy
import sys
from itertools import combinations
import numpy as np
from scipy.optimize import linprog
filename = sys.argv[1] if len(sys.argv) == 2 \
else "in/" + sys.argv[0].split('/')[-1].rstrip(".py") + ".pzl"
data = open(filename).read()
lines = data.rstrip('\n').split('\n')
res1 = 0
res2 = 0
def part1(light, buttons):
for r in range(len(buttons)):
for c in combinations(buttons, r):
state = [0 for _ in light]
for b in c:
for l in b:
state[l] ^= 1
if light == state:
return r
for line in lines:
s = line.split(' ')
s_light = s[0]
s_buttons = s[1:-1]
s_jolt = s[-1]
light = [1 if i == '#' else 0 for i in s_light[1:-1]]
buttons = []
for sb in s_buttons:
buttons.append([int(i) for i in sb[1:-1].split(',')])
jolt = [int(i) for i in s_jolt[1:-1].split(',')]
res = part1(light, buttons)
res1 += res
print('res1:', res1)
print('res2:', res2)
|
