26.06 Распределение объектов по местам (хранение багажа, парковки, кафе, гостиницы)

f = open(’26.txt’)
n = int(f.readline()) # считываем количество пришедших клиентов
k = int(f.readline()) # считываем количество мест в кафе

clients = [] # список клиентов
for line in f.readlines():
    a, b = map(int, line.split())
    clients.append([a, b])
clients.sort() # сортируем список клиентов
table = [-1] * k # список, симулирующий каждое столик, в котором будет записано время, в которое освободится каждое место

c17 = [0]*17 #подсчет количества пассажиров с различными остатками от деления на 17

for client in clients: # проход по каждому клиенту
    ok = False # флаг, которые показывает нужно ли человеку ждать или нет
    for i in range(k):
        # если время прихода текущего клиента больше чем время освобождения данного места другим клиентом
        if client[0] >= table[i]:
            table[i] = client[1] + 20 # записываем время ухода текущего клиента с учётом уборки данного места после него
            ok = True # помечаем, что для текущего клиента нашли место
            tim = client[1]-client[0] #время проведенное за столиком
            c17[tim % 17] += 1 #подсчитываем это время за столиком
            break # прерываем цикл - переходим к следующему клиенту
    if not ok: # если клиенту нужно подождать
        mt = min(table) # определяем место, которое освободится в ближайшее время
        if client[0] + 3 >= mt:  # если время ожидания не более 3 минут
            ind = table.index(mt) # определяем индекс данного места в списке
            tim = client[1] - client[0] # считаем время T
            if table[ind] + tim <= 1200: # если данный клиент уйдёт не позже закрытия
                table[ind] += tim + 20 # то сажаем данного человека за место
                c17[tim % 17] += 1 #подсчитываем это время за столиком

# считаем количество пар, в которых время за столиками кратно 17
pair = c17[0]*(c17[0]-1) // 2
for i in range(1, 9):
    #считаем пары с различными остатками от деления на 17, сумма которых равна 17
    pair += c17[i]*c17[17-i]
print(pair)