25 lines
508 B
Python
25 lines
508 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
def count_trees(area, stepX, stepY):
|
||
|
|
trees = 0
|
||
|
|
posX = 0
|
||
|
|
for i in range(0, len(area), stepY):
|
||
|
|
line = area[i]
|
||
|
|
if line[posX] == '#':
|
||
|
|
trees +=1
|
||
|
|
posX = (posX + stepX) % (len(line))
|
||
|
|
return trees
|
||
|
|
|
||
|
|
area = []
|
||
|
|
with open("input01 .txt","r") as f:
|
||
|
|
for line in f:
|
||
|
|
area.append(line.strip())
|
||
|
|
|
||
|
|
result = 1
|
||
|
|
result *= count_trees(area, 1,1)
|
||
|
|
result *= count_trees(area, 3,1)
|
||
|
|
result *= count_trees(area, 5,1)
|
||
|
|
result *= count_trees(area, 7,1)
|
||
|
|
result *= count_trees(area, 1,2)
|
||
|
|
|
||
|
|
print(result)
|