Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import copy
- #turn the input into an array of ints
- inputFile = open("input.txt","r")
- rawImage = inputFile.read()
- rawImage = list(rawImage)
- for i in range(len(rawImage)):
- rawImage[i] = int(rawImage[i])
- layerSize = int(input("How wide is the image?")) * int(input("How high is the image?"))
- numberOfLayers = len(rawImage)/layerSize
- layers = []
- layerN = []
- #get a two-dimensional array that represents the image at each layer
- for i in range(len(rawImage)):
- layerN.append(rawImage[i])
- if((i+1)%layerSize == 0):
- layers.append(copy.copy(layerN))
- layerN.clear()
- fewestZeros = 150
- selectLayer = []
- #find the layer with the fewest zeros
- for x in layers:
- count = 0
- for y in range(len(x)):
- if(x[y]==0):
- count += 1
- if(count<fewestZeros):
- fewestZeros = count
- selectLayer = x
- print("The layer with the fewest zeros is layer " + str(layers.index(selectLayer)+1) + " with " + str(fewestZeros) + " zeros.")
- ones = 0
- twos = 0
- for i in selectLayer:
- if(i==1):
- ones += 1
- elif(i==2):
- twos += 1
- print("Number of ones in the layer: " + str(ones))
- print("Number of twos in the layer: " + str(twos))
- print("The answer is " + str((ones * twos)))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement