mirror of
https://github.com/Crocmagnon/advent-of-code.git
synced 2024-11-05 14:23:58 +01:00
31 lines
747 B
Python
31 lines
747 B
Python
|
import math
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
width = 25
|
||
|
|
height = 6
|
||
|
|
with open("inputs/day08") as f:
|
||
|
|
data = f.readline().strip()
|
||
|
|
layers = []
|
||
|
|
layer_size = width * height
|
||
|
|
position = 0
|
||
|
|
layer = data[position : position + layer_size]
|
||
|
|
number_of_zero_digits = math.inf
|
||
|
|
layer_index = -1
|
||
|
|
while layer:
|
||
|
|
count = layer.count("0")
|
||
|
|
if count < number_of_zero_digits:
|
||
|
|
layer_index = int(position / layer_size)
|
||
|
|
number_of_zero_digits = count
|
||
|
|
layers.append(layer)
|
||
|
|
position += layer_size
|
||
|
|
layer = data[position : position + layer_size]
|
||
|
|
|
||
|
|
print(layer_index)
|
||
|
|
layer = layers[layer_index]
|
||
|
|
print(layer.count("1") * layer.count("2"))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|