Để tính toán thời gian chờ đợi trung bình cần theo dõi thời gian chờ của mỗi người. Dưới đây là chương trình mở rộng để tính toán thời gian chờ đợi trung bình.

def bepan_with_wait_time(dangkiga, dangkibo, suatan):

    doimon = [] # list chứa ID những người phải đổi món

    wait_times = {}

    current_time = 0

    while not suatan.is_empty():

        current_time += 1  # Mỗi lượt lấy suất ăn là 1 giây

        tmp = suatan.pop()

        if tmp == "bò":  # nếu suất ăn lấy ra là bò

            if not dangkibo.empty():  # còn người đăng kí suất ăn bò

                id = dangkibo.get()

                wait_times[id] = current_time

            else:

                id = dangkiga.get()

                doimon.append(id)

                wait_times[id] = current_time

        elif tmp == "gà":  # nếu suất ăn lấy ra là gà

            if not dangkiga.empty():  # còn người đăng kí suất ăn gà

                id = dangkiga.get()

                wait_times[id] = current_time

            else:

                id = dangkibo.get()

                doimon.append(id)

                wait_times[id] = current_time

    return doimon, wait_times

# Đọc thông tin đăng kí và đưa vào 2 queue

dangkiga = Queue()

dangkibo = Queue()

with open("input1.inp", "r", encoding="utf8") as file1:

    for line in file1.readlines():

        id, dangki = line.strip().split()

        if dangki == "gà":

            dangkiga.put(id)

        elif dangki == "bò":

            dangkibo.put(id)

# Đọc thông tin suất ăn và đưa vào ngăn xếp

suatan = Stack()

with open("input2.inp", "r", encoding="utf8") as file2:

    data = file2.read().strip().split()

    for item in data:

        suatan.push(item)

doimon, wait_times = bepan_with_wait_time(dangkiga, dangkibo, suatan)

if len(doimon) == 0:

    print("Không có người nào phải đổi món ăn")

else:

    print("Danh sách những người phải đổi món là:", doimon)

# Tính toán thời gian chờ đợi trung bình

if wait_times:

    total_wait_time = sum(wait_times.values())

    average_wait_time = total_wait_time / len(wait_times)

    print(f"Thời gian chờ đợi trung bình là: {average_wait_time:.2f} giây")

else: