44 lines
919 B
Python
44 lines
919 B
Python
|
import sys
|
||
|
|
||
|
L = sys.stdin.read().splitlines()
|
||
|
gal = []
|
||
|
S=0
|
||
|
|
||
|
# Parser pour avoir les galaxies
|
||
|
for line, P in enumerate(L):
|
||
|
for col, p in enumerate(P):
|
||
|
if p != ".":
|
||
|
gal.append([line, col])
|
||
|
|
||
|
|
||
|
# Expansion
|
||
|
lines = [l[0] for l in gal]
|
||
|
cols = [l[1] for l in gal]
|
||
|
|
||
|
lines_empty = []
|
||
|
for line in range(0, len(L)):
|
||
|
if lines.count(line) == 0:
|
||
|
lines_empty.append(line)
|
||
|
|
||
|
for line in reversed(lines_empty):
|
||
|
for g in reversed(gal):
|
||
|
if g[0] > line:
|
||
|
g[0] +=999999
|
||
|
|
||
|
cols_empty = []
|
||
|
for col in range(0, len(L[0])):
|
||
|
if cols.count(col) == 0:
|
||
|
cols_empty.append(col)
|
||
|
|
||
|
for col in reversed(cols_empty):
|
||
|
for g in reversed(gal):
|
||
|
if g[1] > col:
|
||
|
g[1] +=999999
|
||
|
|
||
|
|
||
|
# Calcul distance entres les galaxies
|
||
|
for s in range(0, len(gal)):
|
||
|
for n in range(1, len(gal)-s):
|
||
|
S+= (abs(gal[s+n][0] - gal[s][0])) + abs((gal[s+n][1] - gal[s][1]))
|
||
|
|
||
|
print(S)
|