27. Программирование: #298936
Фрагмент звёздного неба спроецирован на плоскость с декартовой системой координат. Учёный решил провести кластеризацию полученных точек, являющихся изображениями звёзд, то есть разбить их множество на $N$ непересекающихся непустых подмножеств (кластеров), таких, что точки каждого подмножества лежат внутри прямоугольника со сторонами длиной $H$ и $W$, причём эти прямоугольники между собой не пересекаются. Стороны прямоугольников не обязательно параллельны координатным осям. Гарантируется, что такое разбиение существует и единственно для заданных размеров прямоугольников.
Будем называть центром кластера точку (звезду) этого кластера, сумма расстояний от которой до всех остальных его точек минимальна. Для каждого кластера гарантируется единственность его центра. Расстояние между двумя точками на плоскости $A(x_1, y_1)$ и $B(x_2, y_2)$ вычисляется по формуле $d(A, B) = \sqrt{(x_2 — x_1)^2 + (y_2 — y_1)^2}$.
Каждая звезда помимо координат на плоской карте характеризуется своим спектральным классом и классом светимости. Спектральный класс определяет цвет (который связан с температурой звезды) следующим образом.
| Обозначение спектрального класса (латинская буква) | $O$ | $B$ | $A$ | $F$ | $G$ | $K$ | $M$ |
|---|---|---|---|---|---|---|---|
| Цвет звезды | Голубой | Бело-голубой | Белый | Жёлто-белый | Жёлтый | Оранжевый | Красный |
Каждый из спектральных классов, в свою очередь, делится на подклассы от $0$ до $9$ в порядке уменьшения температуры. Обозначение подкласса ставится после обозначения спектрального класса (например, $B2$). Класс светимости звезды обозначим римскими цифрами от $I$ до $VII$.
| Обозначение класса светимости | $I$ | $II$ | $III$ | $IV$ | $V$ | $VI$ | $VII$ |
|---|---|---|---|---|---|---|---|
| Светимость | Сверхгигант | Яркий гигант | Гигант | Субгигант | Карлик | Субкарлик | Белый карлик |
В ответе запишите четыре числа через запятую: сначала $A_1$, затем целую часть произведения $A_2 \times 10,000$; сначала целую часть произведения $B_1 \times 10,000$, затем $B_2$.
Пример организации данных в одном из исходных файлов для случая четырёх звёзд
5,01788 8,32466 G2V
4,289251 6,955186 VII
4,619358 5,524697 B7V
6,91934 20,425391 G2V
Внимание! Пример приведён в иллюстративных целях для произвольных значений, не имеющих отношения к заданию. Для выполнения задания используйте данные из прилагаемых файлов.
Сначала расшифруем обозначения по двум таблицам. Жёлтый субгигант — спектральный класс $G$ и светимость $IV$. Оранжевый гигант — класс $K$ и светимость $III$. Белый яркий гигант — класс $A$ и светимость $II$. Классы $V$, $VI$ и $VII$ легко перепутать: $G2V$ — карлик, $G2VI$ — субкарлик, а $VII$ записан вообще без буквы.
Разбор строки. Первые два поля — координаты, третье либо равно $VII$, либо устроено как «буква, цифра, римское число». Значит, спектральный класс — первый символ, класс светимости — всё начиная с третьего (второй символ — подкласс, он нигде не нужен). Белые карлики обрабатываем отдельно: спектрального класса у них нет.
Ещё одна тонкость этих файлов — разнобой в десятичных разделителях: часть чисел записана с точкой и распознана таблицей как число, часть с запятой и осталась текстом. Поэтому надёжнее собрать текст всей строки, заменить запятые на точки и разделить по пробелам — тогда в любом случае получаются три поля.
Кластеризация. Прямоугольники не пересекаются, поэтому кластеры разделены в пространстве, и достаточно разбить точки на компоненты связности: объединяем точки, отстоящие друг от друга меньше чем на порог. Здесь результат устойчив при любом пороге от $0{,}8$ до $2$, что само по себе хороший признак. Проверяем: число кластеров совпадает с условием, и каждый укладывается в прямоугольник $6{,}0 \times 5{,}5$.
Центр кластера — точка самого кластера с минимальной суммой расстояний до остальных, ищем перебором. Считать среднее арифметическое координат нельзя.
Обратите внимание на разную логику двух величин: в $A_1$ и $B_1$ речь идёт об удалении от центра своего кластера, но в первом случае считается количество звёзд в круге радиуса $1{,}2$, а во втором — наибольшее удаление, причём максимум берётся по всем кластерам сразу.
import zipfile, math
from xml.etree import ElementTree as ET
T = '{urn:oasis:names:tc:opendocument:xmlns:table:1.0}'
O = '{urn:oasis:names:tc:opendocument:xmlns:office:1.0}'
def load(name):
x = ET.fromstring(zipfile.ZipFile(name).read('content.xml'))
P = []
for r in x.iter(T + 'table-row'):
parts = []
for c in r.iter(T + 'table-cell'):
v = c.get(O + 'value') or ''.join(c.itertext()).strip()
if v: parts.append(v)
tok = ' '.join(parts).replace(',', '.').split() # чиним разнобой форматов
if len(tok) < 3: continue
s = tok[2]
sp, lum = (None, 'VII') if s == 'VII' else (s[0], s[2:])
P.append((float(tok[0]), float(tok[1]), sp, lum))
return P
def center(c):
return min(c, key=lambda a: sum(math.dist(a[:2], b[:2]) for b in c))
def clusters(pts, eps=1.0):
unused, res = set(range(len(pts))), []
while unused:
i = unused.pop(); comp, stack = [i], [i]
while stack:
a = stack.pop()
for j in list(unused):
if math.dist(pts[a][:2], pts[j][:2]) < eps:
unused.discard(j); comp.append(j); stack.append(j)
res.append([pts[k] for k in comp])
return res
A = load('27-11_A.ods')
ca = clusters(A)
small = min(ca, key=len) # наименьший кластер
cn = center(small)
A1 = sum(p[2] == 'G' and p[3] == 'IV' and math.dist(p[:2], cn[:2]) <= 1.2
for p in small)
A2 = math.dist(center(ca[0])[:2], center(ca[1])[:2])
print(A1, int(A2 * 10000))
B = load('27-11_B.ods')
B1 = B2 = 0
for c in clusters(B):
cc = center(c)
for p in c:
if p[2] == 'K' and p[3] == 'III':
B1 = max(B1, math.dist(p[:2], cc[:2]))
B2 = max(B2, sum(p[2] == 'A' and p[3] == 'II' for p in c))
print(int(B1 * 10000), B2)
Результаты для файла $А$. Всего $457$ звёзд, кластеры содержат $219$ и $238$ точек. Наименьший — из $219$ звёзд, он занимает область $x \in [2{,}94;\ 7{,}41]$, $y \in [5{,}43;\ 9{,}21]$, его центр — звезда $M3III$ с координатами $(5{,}025911;\ 7{,}299077)$. Жёлтых субгигантов ($G$, $IV$) в этом кластере $21$, в круг радиуса $1{,}2$ вокруг центра попадают $14$ из них, значит, $A_1 = 14$.
Центр второго кластера — $(7{,}878035;\ 19{,}907808)$, расстояние между центрами $A_2 = 12{,}9272854$, что даёт $129272$.
Результаты для файла $Б$. Всего $1230$ звёзд, кластеры содержат $887$, $199$ и $144$ точки с центрами $(26{,}240133;\ 17{,}935236)$, $(16{,}355297;\ 19{,}37937)$ и $(14{,}592412;\ 26{,}897683)$. Оранжевых гигантов ($K$, $III$) в них $59$, $20$ и $14$; наибольшее удаление такой звезды от центра своего кластера по кластерам равно $2{,}2690$, $1{,}9855$ и $1{,}6737$, значит, $B_1 = 2{,}268970$ и в ответ идёт $22689$.
Белых ярких гигантов ($A$, $II$) в кластерах $21$, $3$ и $1$, поэтому $B_2 = 21$.