114 lines
2.9 KiB
Python
114 lines
2.9 KiB
Python
|
from solutions import BaseSolution
|
||
|
|
|
||
|
|
|
||
|
|
class Solution(BaseSolution):
|
||
|
|
input_file = "08.txt"
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return "Day 8: Seven Segment Search"
|
||
|
|
|
||
|
|
def parse_input(self, data):
|
||
|
|
def parse(l):
|
||
|
|
i, o = l.split(" | ")
|
||
|
|
return i.split(), o.split()
|
||
|
|
|
||
|
|
return [parse(l) for l in data.strip().splitlines()]
|
||
|
|
|
||
|
|
def solve(self, puzzle_input):
|
||
|
|
return sum(sum(len(d) in (2, 3, 4, 7) for d in l[1]) for l in puzzle_input)
|
||
|
|
|
||
|
|
def solve_again(self, puzzle_input):
|
||
|
|
digits = [
|
||
|
|
"abcefg",
|
||
|
|
"cf",
|
||
|
|
"acdeg",
|
||
|
|
"acdfg",
|
||
|
|
"bcdf",
|
||
|
|
"abdfg",
|
||
|
|
"abdefg",
|
||
|
|
"acf",
|
||
|
|
"abcdefg",
|
||
|
|
"abcdfg",
|
||
|
|
]
|
||
|
|
outs = []
|
||
|
|
|
||
|
|
def outsum(n):
|
||
|
|
out = ["".join(sorted(n[s] for s in d)) for d in digits]
|
||
|
|
outs.append(
|
||
|
|
int(
|
||
|
|
"".join(
|
||
|
|
map(
|
||
|
|
str,
|
||
|
|
[out.index(i) for i in ["".join(sorted(oo)) for oo in o]],
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
for i, o in puzzle_input:
|
||
|
|
n = {}
|
||
|
|
v1, v7, v4, v8 = sorted(
|
||
|
|
list(filter(lambda o: len(o) in (2, 3, 4, 7), i)), key=lambda x: len(x)
|
||
|
|
)
|
||
|
|
c, f = v1
|
||
|
|
n["a"] = a = (set(v7) - set(v1)).pop()
|
||
|
|
v235 = list(filter(lambda d: len(d) == 5, i))
|
||
|
|
for v in v235:
|
||
|
|
diff = set(v) - set(v4 + a)
|
||
|
|
if len(diff) == 1:
|
||
|
|
n["g"] = g = diff.pop()
|
||
|
|
break
|
||
|
|
n["c"] = c
|
||
|
|
n["f"] = f
|
||
|
|
n["e"] = e = (set(v8) - set(v4 + a + g)).pop()
|
||
|
|
for v in v235:
|
||
|
|
diff = set(v) - set(n.values())
|
||
|
|
if len(diff) == 1:
|
||
|
|
n["d"] = d = diff.pop()
|
||
|
|
break
|
||
|
|
n["b"] = b = (set(v4) - set([d, c, f])).pop()
|
||
|
|
|
||
|
|
try:
|
||
|
|
outsum(n)
|
||
|
|
except ValueError:
|
||
|
|
n["c"] = f
|
||
|
|
n["f"] = c
|
||
|
|
outsum(n)
|
||
|
|
|
||
|
|
return sum(outs)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
solution = Solution()
|
||
|
|
solution.show_results()
|
||
|
|
|
||
|
|
"""
|
||
|
|
dddd
|
||
|
|
e a
|
||
|
|
e a
|
||
|
|
ffff
|
||
|
|
g b
|
||
|
|
g b
|
||
|
|
cccc
|
||
|
|
|
||
|
|
|
||
|
|
0: 1: 2: 3: 4:
|
||
|
|
aaaa .... aaaa aaaa ....
|
||
|
|
b c . c . c . c b c
|
||
|
|
b c . c . c . c b c
|
||
|
|
.... .... dddd dddd dddd
|
||
|
|
e f . f e . . f . f
|
||
|
|
e f . f e . . f . f
|
||
|
|
gggg .... gggg gggg ....
|
||
|
|
|
||
|
|
5: 6: 7: 8: 9:
|
||
|
|
aaaa aaaa aaaa aaaa aaaa
|
||
|
|
b . b . . c b c b c
|
||
|
|
b . b . . c b c b c
|
||
|
|
dddd dddd .... dddd dddd
|
||
|
|
. f e f . f e f . f
|
||
|
|
. f e f . f e f . f
|
||
|
|
gggg gggg .... gggg gggg
|
||
|
|
|
||
|
|
"""
|