Skip to content

Instantly share code, notes, and snippets.

@w4
Created September 8, 2016 08:14
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save w4/95af360bfaa0f5905a45a15851bdbe1d to your computer and use it in GitHub Desktop.
Save w4/95af360bfaa0f5905a45a15851bdbe1d to your computer and use it in GitHub Desktop.
from functools import reduce
import operator
grid = [
[8, 49, 81, 52, 22, 24, 32, 67, 24, 21, 78, 16, 86, 19, 4, 88, 4, 20, 20, 1],
[2, 49, 49, 70, 31, 47, 98, 26, 55, 36, 17, 39, 56, 80, 52, 36, 42, 69, 73, 70],
[22, 99, 31, 95, 16, 32, 81, 20, 58, 23, 53, 5, 0, 81, 8, 68, 16, 36, 35, 54],
[97, 40, 73, 23, 71, 60, 28, 68, 5, 9, 28, 42, 48, 68, 83, 87, 73, 41, 29, 71],
[38, 17, 55, 4, 51, 99, 64, 2, 66, 75, 22, 96, 35, 5, 97, 57, 38, 72, 78, 83],
[15, 81, 79, 60, 67, 3, 23, 62, 73, 0, 75, 35, 71, 94, 35, 62, 25, 30, 31, 51],
[0, 18, 14, 11, 63, 45, 67, 12, 99, 76, 31, 31, 89, 47, 99, 20, 39, 23, 90, 54],
[40, 57, 29, 42, 89, 2, 10, 20, 26, 44, 67, 47, 7, 69, 16, 72, 11, 88, 1, 69],
[0, 60, 93, 69, 41, 44, 26, 95, 97, 20, 15, 55, 5, 28, 7, 3, 24, 34, 74, 16],
[75, 87, 71, 24, 92, 75, 38, 63, 17, 45, 94, 58, 44, 73, 97, 46, 94, 62, 31, 92],
[4, 17, 40, 68, 36, 33, 40, 94, 78, 35, 3, 88, 44, 92, 57, 33, 72, 99, 49, 33],
[5, 40, 67, 56, 54, 53, 67, 39, 78, 14, 80, 24, 37, 13, 32, 67, 18, 69, 71, 48],
[7, 98, 53, 1, 22, 78, 59, 63, 96, 0, 4, 0, 44, 86, 16, 46, 8, 82, 48, 61],
[78, 43, 88, 32, 40, 36, 54, 8, 83, 61, 62, 17, 60, 52, 26, 55, 46, 67, 86, 43],
[52, 69, 30, 56, 40, 84, 70, 40, 14, 33, 16, 54, 21, 17, 26, 12, 29, 59, 81, 52],
[12, 48, 3, 71, 28, 20, 66, 91, 88, 97, 14, 24, 58, 77, 79, 32, 32, 85, 16, 1],
[50, 4, 49, 37, 66, 35, 18, 66, 34, 34, 9, 36, 51, 4, 33, 63, 40, 74, 23, 89],
[77, 56, 13, 2, 33, 17, 38, 49, 89, 31, 53, 29, 54, 89, 27, 93, 62, 4, 57, 19],
[91, 62, 36, 36, 13, 12, 64, 94, 63, 33, 56, 85, 17, 55, 98, 53, 76, 36, 5, 67],
[8, 0, 65, 91, 80, 50, 70, 21, 72, 95, 92, 57, 58, 40, 66, 69, 36, 16, 54, 48]
]
biggest = 0
used = []
for y, row in enumerate(grid):
for x, column in enumerate(row):
left = reduce(operator.mul, grid[y][x - 4:x], 1)
used = grid[y][x - 4:x] if left > biggest else used
biggest = left if left > biggest else biggest
right = reduce(operator.mul, grid[y][x:x + 4])
used = grid[y][x:x + 4] if right > biggest else used
biggest = right if right > biggest else biggest
down = reduce(operator.mul, [n[x] for n in grid[y:y + 4]], 1)
used = [n[x] for n in grid[y:y + 4]] if down > biggest else used
biggest = down if down > biggest else biggest
up = reduce(operator.mul, [n[x] for n in grid[y - 4:y]], 1)
used = [n[x] for n in grid[y - 4:y]] if up > biggest else used
biggest = up if up > biggest else biggest
if x + 4 < 20:
values = [n[x + i] for i, n in enumerate(grid[y - 4:y])]
forwardDiag = reduce(operator.mul, values, 1)
used = values if forwardDiag > biggest else used
biggest = forwardDiag if forwardDiag > biggest else biggest
values = [n[x - i] if x - i > 0 else 0 for i, n in enumerate(grid[y:y + 4])]
backwardDiag = reduce(operator.mul, values, 1)
used = values if backwardDiag > biggest else used
biggest = backwardDiag if backwardDiag > biggest else biggest
print(biggest, used)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment