tetrika_tasks/task_3.py

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"]}'