🐣 상담원 인원

1. 문제 링크

상담원 인원

2. 코드

# FIFO
# 각 멘토당 끝나는 시간 기록
# 끝나는 시간이 더 빠른 멘토에게 상담 진행
# 우선 멘토 한명씩 배정하고 이후, 기다리는 시간이 가장 긴 팀에게 멘토 추가 배정

def min_time(start_time, end_time, group_queue):
    wait_time = 0
    min_v = int(1e9)
    target_idx = 0

    for i in range(len(group_queue)):
        if group_queue[i] <= start_time:
            target_idx = i
            min_v = 0
            break
        # 시간차가 최소가 되는 값 찾기
        if min_v > group_queue[i] - start_time:
            target_idx = i
            min_v = group_queue[i] - start_time

    group_queue[target_idx] = max(start_time, group_queue[target_idx]) + end_time # 끝나는 시간으로 바꿔줌
    wait_time = min_v

    return wait_time, group_queue


def solution(k, n, reqs):
    answer = 0
    wait_queue = [[0] for _ in range(k + 1)]
    n = n - k

    while n >= 0:
        wait_time = [0] * (k + 1) # 각 그룹별로 기다려야 하는 시간 총합
        for start, end, g in reqs:
            t, q = min_time(start, end, wait_queue[g])
            wait_time[g] += t
            wait_queue[g] = q

        max_time = max(wait_time) # 각 그룹 중 최대로 기다려야하는 그룹의 시간
        wait_queue[wait_time.index(max_time)].append(0)
        n -= 1
        for i in range(k + 1):
            wait_queue[i] = len(wait_queue[i]) * [0]

    answer = sum(wait_time)
    return answer