Задача к ЕГЭ по информатике на тему «Анализ данных (звезды)» №16

Учёный решил провести кластеризацию некоторого множества звёзд по их расположению на карте звёздного неба. Кластер звёзд – это набор звёзд (точек) на графике, каждая из которых находится от хотя бы одной другой звезды на расстоянии не более $R$ условных единиц. Каждая звезда обязательно принадлежит только одному из кластеров.

Двойная звездная система – это две звезды на расстоянии не более $t$ . При этом других звезд на расстоянии менее $t$ у этих двух звезд быть не должно.

Под расстоянием понимается расстояние Евклида между двумя точками $A(x1,y1)$ и $B(x2,y2)$ на плоскости, которое вычисляется по формуле:

∘ -------------------- d(A, B) = (x2 − x1)2 + (y2 − y1)2

Аномалиями назовём точки, находящиеся на расстоянии более одной условной единицы от точек кластеров. При расчётах аномалии учитывать не нужно.

В файле A хранятся данные о звёздах пяти кластеров, где $R = 0.9$ , $t = 0.2$ для каждого кластера. В каждой строке записана информация о расположении на карте одной звезды, а также ее масса (в солнечных массах): сначала координата $x$ , затем координата $y$ , затем масса $m$ . Значения даны в условных единицах, которые представлены вещественными числами. Известно, что количество звёзд не превышает 1000.

В файле Б хранятся данные о звёздах четырёх кластеров, где $R = 0.9$ , $t = 0.02$ для каждого кластера. Известно, что количество звёзд не превышает 10 000. Структура хранения информации о звездах в файле Б аналогична файлу А.

Для каждого файла в каждом кластере найдите двойную звездную систему, состоящую из красного карлика (масса от 0.08 до 0.6 солнечных масс) и желтого карлика (масса от 0.8 до 1.2 солнечных масс) с максимальным расстоянием между ними. Затем вычислите два числа: $P x$ – среднее арифметическое абсцисс найденных звезд, и $P y$ – среднее арифметическое ординат найденных звезд.

В ответе запишите четыре числа через пробел: сначала целую часть произведения $Px ⋅1000$ для файла А, затем $Py ⋅1000$ для файла А, далее целую часть произведения $Px ⋅1000$ для файла Б и $Py ⋅1000$ для файла Б.

Возможные данные одного из файлов иллюстрированы графиком.

Внимание! График приведён в иллюстративных целях для произвольных значений, не имеющих отношения к заданию. Для выполнения задания используйте данные из прилагаемого файла.

Для разделения звезд на кластеры будем использовать функцию dbscan.

Дальше основная идея решения будет заключаться в том, что мы будем проходить по каждой точке в пяти для файла А и четырех для файла Б найденных кластерах и с помощью того же метода dbscan для каждого кластера найти списки звезд, расстояние между которыми менее 0.3 для файла А и 0.03 для файла Б.

В каждом кластере нужно оставить только те списки, в которых количество звезд равно двум – то есть только двойные звездные системы. Также нужно проверить по массе звезд, чтобы в паре были красный и желтый карлики.

В конце остается дело за малым: для каждой звездной системы в каждом кластере найти пару с максимальным расстоянием между звездами, а затем рассчитать среднее расстояние между всеми найденными парами.

Диаграмма для файла А имеет вид:

Диаграмма для файла Б имеет вид:

Программа:

from math import *
def dbscan(a, r):
    cl = [] # Инициализируем список для хранения кластеров
    while a: # Пока есть элементы в входном массиве ’a’
        # Создаем новый кластер и добавляем в него первый элемент из ’a’
        cl.append([a.pop(0)])
        for i in cl[-1]: # Проходим по элементам последнего кластера
            # Проверяем  каждый элемент ’j’ в оставшихся элементах ’a’
            for j in a[:]:
                # Если расстояние между ’i’ и ’j’ меньше радиуса ’r’
                x = [i[0], i[1]]
                y = [j[0], j[1]]
                if dist(x, y) <= r:
                    cl[-1].append(j) # Добавляем ’j’ в текущий кластер
                    a.remove(j) # Удаляем ’j’ из списка ’a’, чтобы не проверять его снова
    return cl

f = open("2.txt")
s = f.readline()
a = [list(map(float, i.replace(",", ".").split())) for i in f]
cl = dbscan(a, 0.9)
cl_total = []
for i in cl:
    if len(i) > 10: cl_total.append(i)

t = 0.2 # Для файла А
t = 0.02 # Для файла Б
ans = []
for i in cl_total: # Проходим по каждому элементу в списке cl_total
    found_star = dbscan(i, t) # Применяем алгоритм DBSCAN
    bin_stars = [] # Список для двойных звездных систем
    mx_starsys = [] # Список для хранения звездной системы с максимальным расстоянием
    # Проходим по каждому кластеру, найденному алгоритмом DBSCAN
    for j in found_star:
        if len(j) == 2: # Проверяем, состоит ли кластер из двух звезд
            # Проверяем есть ли в системе красный и желтый карлики
            if ((0.08 <= j[0][2] <= 0.6 and 0.8 <= j[1][2] <= 1.2)
            or (0.08 <= j[1][2] <= 0.6 and 0.8 <= j[0][2] <= 1.2)):
                bin_stars.append(j)
    mx_dist = 0 # Переменная для хранения максимального расстояния
    for j in bin_stars: # Проходим по всем найденным двойным системам
        x = [j[0][0], j[0][1]]






































































































































































































        y = [j[1][0], j[1][1]]
        # Вычисляем периметр
        if dist(x, y) > mx_dist:
            mx_dist = dist(x, y)  # Обновляем максимальное расстояние
            mx_starsys = j # Сохраняем текущую звездную систему
    ans.append(mx_starsys)

res_X = 0
res_Y = 0
for i in ans:
    res_X += (i[0][0] + i[1][0])
    res_Y += (i[0][1] + i[1][1])

print(int(res_X / (5 * 2) * 1000))  # Для файла А
print(int(res_Y / (5 * 2) * 1000))  # Для файла А

print(int(res_X / (4 * 2) * 1000))  # Для файла Б
print(int(res_Y / (4 * 2) * 1000))  # Для файла Б

Ответ: 9521 -5587 4083 2186