1234567891011121314151617181920212223242526272829303132333435 |
- import argparse
- import functools
- from typing import NamedTuple, List, Dict, Set
- parser = argparse.ArgumentParser()
- parser.add_argument("ifile", type=argparse.FileType('r'))
- args = parser.parse_args()
- heightmap = [[int(x) for x in line.strip()] for line in args.ifile.readlines()]
- def printMap(map: List[List]):
- for line in map:
- print("".join([str(int(el)) for el in line]))
- def descendingMap(seq: int) -> List[bool]:
- return [True] + list(map(lambda pair: pair[0] < pair[1], zip(seq[1:], seq)))
- right = map(descendingMap, heightmap)
- left = map(lambda line: reversed(descendingMap(list(reversed(line)))), heightmap)
- horiz = [[all(pair) for pair in line] for line in map(lambda pair: zip(pair[0], pair[1]), zip(right, left))]
- # transpose array to check cols
- flipped = [[heightmap[j][i] for j in range(len(heightmap))] for i in range(len(heightmap[0]))]
- up = map(descendingMap, flipped)
- down = map(lambda line: reversed(descendingMap(list(reversed(line)))), flipped)
- pre_vert = [[all(pair) for pair in line] for line in map(lambda pair: zip(pair[0], pair[1]), zip(up, down))]
- # transpose back
- vert = [[pre_vert[j][i] for j in range(len(pre_vert))] for i in range(len(pre_vert[0]))]
- composite = [[all(pair) for pair in line] for line in map(lambda pair: zip(pair[0], pair[1]), zip(horiz, vert))]
- paired = [[pair[0] + 1 if pair[1] else 0 for pair in line] for line in map(lambda pair: zip(pair[0], pair[1]), zip(heightmap, composite))]
- print(sum(map(sum, paired)))
|