84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
||
|
||
import sys
|
||
|
||
input_file = sys.argv[1]
|
||
|
||
with open(input_file) as fd:
|
||
lines = [line.rstrip() for line in fd.readlines()]
|
||
|
||
height = len(lines)
|
||
width = len(lines[0])
|
||
|
||
visited: set[tuple[int, int]] = set()
|
||
|
||
directions = [
|
||
(0, 1),
|
||
(0, -1),
|
||
(1, 0),
|
||
(-1, 0),
|
||
]
|
||
|
||
|
||
def get_region(
|
||
pos: tuple[int, int], region: set[tuple[int, int]]
|
||
) -> set[tuple[int, int]]:
|
||
region.add(pos)
|
||
i, j = pos
|
||
ochar = lines[i][j]
|
||
for direction in directions:
|
||
i, j = pos[0] + direction[0], pos[1] + direction[1]
|
||
if i not in range(height) or j not in range(width):
|
||
continue
|
||
if (i, j) in region:
|
||
continue
|
||
char = lines[i][j]
|
||
if char != ochar:
|
||
continue
|
||
region |= get_region((i, j), region)
|
||
return region
|
||
|
||
|
||
def get_perimeter(region: set[tuple[int, int]]) -> int:
|
||
peri = 0
|
||
for axis in (0, 1):
|
||
oaxis = 0 if axis else 1
|
||
iss = set([pos[axis] for pos in region])
|
||
print(47, iss, peri)
|
||
for i in iss:
|
||
line = [pos[oaxis] for pos in region if pos[axis] == i]
|
||
line.sort()
|
||
last_j = None
|
||
for j in line:
|
||
if last_j is None:
|
||
peri += 1
|
||
elif last_j == j - 1:
|
||
pass
|
||
else:
|
||
peri += 2
|
||
last_j = j
|
||
if last_j is not None:
|
||
peri += 1
|
||
print(62, i, peri, line)
|
||
return peri
|
||
|
||
|
||
tprice = 0
|
||
for i in range(height):
|
||
for j in range(width):
|
||
pos = i, j
|
||
if pos in visited:
|
||
continue
|
||
region = get_region(pos, set())
|
||
visited |= region
|
||
|
||
area = len(region)
|
||
peri = get_perimeter(region)
|
||
price = area * peri
|
||
tprice += price
|
||
|
||
char = lines[i][j]
|
||
print(f"{char}: {area} × {peri} = {price}$")
|
||
|
||
print(tprice)
|