Задача #3884
Анализ данных
Фрагмент звёздного неба спроецирован на плоскость с декартовой системой координат. Учёный решил провести кластеризацию полученных точек, являющихся изображениями звёзд, то есть разбить их множество на N непересекающихся непустых подмножеств (кластеров), таких что точки каждого подмножества лежат внутри прямоугольника со сторонами длиной H и W, причём эти прямоугольники между собой не пересекаются. Стороны прямоугольников не обязательно параллельны координатным осям.
Гарантируется, что такое разбиение существует и единственно для заданных размеров прямоугольников.
Будем называть центром кластера точку этого кластера, сумма расстояний от которой до всех остальных точек кластера минимальна. Для каждого кластера гарантируется единственность его центра. Расстояние между двумя точками на плоскости и вычисляется по формуле:
Каждая звезда помимо координат на плоской карте характеризуется своим спектральным классом и классом светимости. Спектральный класс определяет цвет (который связан с температурой звезды) согласно таблице. Каждый из спектральных классов, в свою очередь, делится на подклассы от 0 до 9 в порядке уменьшения температуры. Обозначение подкласса ставится после обозначения спектрального класса (например, B2). Класс светимости звезды обозначим римскими цифрами от I до VII.
| Обозначение | Цвет | Обозначение | Размер |
| G | белый | I | сверхгигант |
| J | зелёный | II | яркий гигант |
| L | синий | III | гигант |
| N | оранжевый | IV | субгигант |
| Y | красный | V | карлик |
| S | голубой | VI | субкарлик |
| Z | жёлтый | VII | белый карлик |
В файле A хранятся данные о звёздах двух кластеров, где H=6,5, W=4,5 для каждого кластера. каждой строке записана информация о расположении на карте одной звезды: сначала координата x, затем координата y. Далее в той же строке для звёзд классов светимости I–VI указываются спектральный класс, подкласс и класс светимости. Обозначения классов ничем не разделяются. Для звёзд класса светимости VII (Белый карлик) обозначения спектрального класса и подкласса в файле не указываются. Известно, что количество точек не превышает 1000.
В файле B хранятся данные о звёздах трёх кластеров, где H=6,5, W=5 для каждого кластера. Известно, что количество звёзд не превышает 10 000. Структура хранения информации о звездах в файле B аналогична файлу А.
Для файла А определите координаты центра каждого кластера, затем найдите два числа: – абсцисса центра кластера с минимальным количеством гигантов, и – ордината центра кластера с максимальным количеством гигантов.
Для файла Б определите координаты центра каждого кластера, затем найдите два числа: – минимальное расстояние между зелёными звёздами, принадлежащими двум различным кластерам, и – максимальное расстояние между зелёными звёздами, принадлежащими двум различным кластерам.
В ответе запишите четыре числа: в первой строке – сначала целую часть произведения × 10 000, затем целую часть произведения × 10 000; во второй строке – сначала целую часть произведения × 10 000, затем целую часть произведения × 10 000.
Внимание! График приведён в иллюстративных целях для произвольных значений, не имеющих отношения к заданию.
Для выполнения задания используйте данные из прилагаемого файла.

Решение
Ответ
A = [[],[]]
for s in open('27_A.txt'):
x,y,t = s.replace(',','.').split()
x,y = float(x), float(y)
if t=='VII': t=' VII'
if y>10: A[0].append([x,y,t])
else: A[1].append([x,y,t])
B = [[],[],[]]
for s in open('27_B.txt'):
x,y,t = s.replace(',','.').split()
x,y = float(x), float(y)
if t=='VII': t=' VII'
if y>23: B[0].append([x,y,t])
elif y>15: B[1].append([x,y,t])
else: B[2].append([x,y,t])
def dist(p1,p2):
x1,y1,t1 = p1
x2,y2,t2 = p2
return ((x2-x1)**2 + (y2-y1)**2)**0.5
def centr(cl):
m = []
for p in cl:
s = sum(dist(p,p1) for p1 in cl)
m.append([s,p])
return min(m)[1]
k0 = len([p for p in A[0] if p[2][2:]=='III'])
k1 = len([p for p in A[1] if p[2][2:]=='III'])
#print(k0,k1)
x0,y0,t0 = centr(A[0])
x1,y1,t1 = centr(A[1])
A1 = int(x0*10000)
A2 = int(y1*10000)
print(A1, A2)
k0 = [p for p in B[0] if p[2][0]=='J']
k1 = [p for p in B[1] if p[2][0]=='J']
k2 = [p for p in B[2] if p[2][0]=='J']
r = [dist(p1,p2) for p1 in k0 for p2 in k1+k2]+\
[dist(p1,p2) for p1 in k1 for p2 in k2]
B1 = int(min(r)*10000)
B2 = int(max(r)*10000)
print(B1, B2)