47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
def part1(inp):
|
||
|
|
total = 0
|
||
|
|
for display in inp:
|
||
|
|
_, output_values = display
|
||
|
|
for value in output_values:
|
||
|
|
if len(value) in [2, 4, 3, 7]:
|
||
|
|
total += 1
|
||
|
|
print(f"Part 1 : {total}")
|
||
|
|
|
||
|
|
|
||
|
|
def part2(inp):
|
||
|
|
for display in inp:
|
||
|
|
patterns, values = display
|
||
|
|
patterns = sorted(patterns, key=lambda x: len(x))
|
||
|
|
print(patterns)
|
||
|
|
# easy
|
||
|
|
d1 = [x for x in patterns if len(x) == 2][0]
|
||
|
|
print("1", d1)
|
||
|
|
d4 = [x for x in patterns if len(x) == 4][0]
|
||
|
|
print("4", d4)
|
||
|
|
d7 = [x for x in patterns if len(x) == 3][0]
|
||
|
|
print("7", d7)
|
||
|
|
d8 = [x for x in patterns if len(x) == 7][0]
|
||
|
|
print("8", d8)
|
||
|
|
|
||
|
|
# 3 is the only digit that has all common segments with 1
|
||
|
|
breakpoint()
|
||
|
|
d3 = [x for x in patterns if set(d1).issubset(set(x)) and len(x) == 5][0]
|
||
|
|
print("3", d3)
|
||
|
|
|
||
|
|
break
|
||
|
|
|
||
|
|
def main(infile):
|
||
|
|
inp = []
|
||
|
|
with open(infile) as f:
|
||
|
|
for display in f:
|
||
|
|
display = display.rstrip().split(" | ")
|
||
|
|
signal_patterns = display[0].split(" ")
|
||
|
|
output_values = display[1].split(" ")
|
||
|
|
inp.append([signal_patterns, output_values])
|
||
|
|
part1(inp)
|
||
|
|
part2(inp)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main("input.txt")
|