a.py (538B) [raw]
1 #!/usr/bin/env python3 2 import sys 3 4 grid = [] 5 for l in sys.stdin: 6 grid.append([int(x) for x in l.strip()]) 7 8 9 def lowest(r, c): 10 pt = grid[r][c] 11 if r > 0 and grid[r - 1][c] <= pt: 12 return False 13 if c > 0 and grid[r][c - 1] <= pt: 14 return False 15 if c < 99 and grid[r][c + 1] <= pt: 16 return False 17 if r < 99 and grid[r + 1][c] <= pt: 18 return False 19 return True 20 21 22 risk = 0 23 for r in range(100): 24 for c in range(100): 25 if lowest(r, c): 26 risk += grid[r][c] + 1 27 28 print(risk)