tests = [ { "data": { "lesson": [1594663200, 1594666800], "pupil": [ 1594663340, 1594663389, 1594663390, 1594663395, 1594663396, 1594666472, ], "tutor": [1594663290, 1594663430, 1594663443, 1594666473], }, "answer": 3117, }, { "data": { "lesson": [1594702800, 1594706400], "pupil": [ 1594702789, 1594704500, 1594702807, 1594704542, 1594704512, 1594704513, 1594704564, 1594705150, 1594704581, 1594704582, 1594704734, 1594705009, 1594705095, 1594705096, 1594705106, 1594706480, 1594705158, 1594705773, 1594705849, 1594706480, 1594706500, 1594706875, 1594706502, 1594706503, 1594706524, 1594706524, 1594706579, 1594706641, ], "tutor": [ 1594700035, 1594700364, 1594702749, 1594705148, 1594705149, 1594706463, ], }, "answer": 3577, }, { "data": { "lesson": [1594692000, 1594695600], "pupil": [1594692033, 1594696347], "tutor": [1594692017, 1594692066, 1594692068, 1594696341], }, "answer": 3565, }, ] def find_intersection(interval_A, interval_B) -> list: # если один из интервалов пустой, то пересечение будет пустым if not interval_A or not interval_B: return [] # если какой-то из инетрвалов имеет нулевую длину (начало=конец), # то пересечение пустое if interval_A[0] == interval_A[1] or interval_B[0] == interval_B[1]: return [] # нужно проверить, что интервалы вообще пересекаются # А и B не пересекаются, если # начало А > конец B, или если начало B > конец А if interval_A[0] > interval_B[1] or interval_B[0] > interval_A[1]: return [] # начало интервала = наименьший из начал интервалов А и B # конец интервала это наименьший из концов интервалов А и B interval_begin = max(interval_A[0], interval_B[0]) interval_end = min(interval_A[1], interval_B[1]) return [interval_begin, interval_end] def enlarge_elapsed_intervals(intervals): # ключ - начало, значние - конец. Если по ключу "начало" уже есть # значение, то сохраняем max(существующее, новое) # чистка конечных интервалов. ключ в словаре - начало интервала. # проходим по исходному списку и выбираем наибольшее конечное # значение (самый поздний) из интервалов с одинаковыми началами beginnings = {interval[0]: interval[1] for interval in intervals} for interval in intervals: begin = interval[0] end = interval[1] beginnings[begin] = max(beginnings[begin], end) print(f"{beginnings=}") endings = {beginnings[k]: k for k in beginnings} for interval in intervals: begin = interval[0] end = interval[1] if endings.get(end): endings[end] = min(endings[end], begin) print(f"{endings=}") result = [[endings[k], k] for k in endings] return result def appearance(intervals): lesson = intervals["lesson"] pupil = intervals["pupil"] tutor = intervals["tutor"] pupil2 = [pupil[i : i + 2] for i in range(0, len(pupil), 2)] # print(f"{pupil2=}") # pupil2 = enlarge_elapsed_intervals(pupil2) enlarge_elapsed_intervals(test) # получаем все интервалы студента на уроке pupil_on_lesson = [ find_intersection(lesson, pupil[i : i + 2]) for i in range(0, len(pupil), 2) ] print(f"{pupil_on_lesson=}") # по очереди сравниваем каждый из интервалов преподавателя с # каждым из интервалов пересечения студента на уроке tutor_on_pupil = [ find_intersection(pol, tutor[i : i + 2]) for i in range(0, len(tutor), 2) for pol in pupil_on_lesson ] print(f"{tutor_on_pupil=}") # создаем список из продолжительности каждого интервала, если он # не пустой, а затем суммируем все элементы answer = sum([intr[1] - intr[0] for intr in tutor_on_pupil if intr]) print("----------------------------------------") return answer if __name__ == "__main__": for i, test in enumerate(tests): test_answer = appearance(test["data"]) assert ( test_answer == test["answer"] ), f'Error on test case {i}, got {test_answer}, expected {test["answer"]}'