🦊 상담원 인원

1. 문제 링크

상담원 인원

2. 코드

from itertools import product

def solution(k, n, reqs):
    mentor = range(1, n + 1)
    mentors = [p for p in product(mentor, repeat=k) if sum(p) == n]
    
    min_wait_time = 10000000
    # 멘토 배치하는 모든 경우의 수
    for mentor_case in mentors:
        # 상담 대기큐
        wating_queue = []
        for i in range(k):
            wating_queue.append([])
        # n번째 케이스 검사
        for i in range(len(mentor_case)):
            # i+1 번 멘토의 걸리는 시간 계산
            mentor_index = i + 1
            mentor_count = mentor_case[i] # 멘토 수
            total_wait_time = 0
            for start, time, number in reqs:
                # 이 멘토에 해당하는 상담 케이스면
                if number == mentor_index:
                    # 상담 큐 검사
                    # 아직 사람을 받을 수 있으면 저장
                    if len(wating_queue[i]) < mentor_count:
                        wating_queue[i].append([start, start + time])
                    # 사람을 못받으면 앞에 것을 빼내고 시간 검사
                    else:
                        # 제일 빨리 끝나는 순으로 저장
                        wating_queue[i] = sorted(wating_queue[i], key=lambda x: x[1])
                        started, end = wating_queue[i].pop(0)
                        # 시작 시간이 제일 빨리 끝난 사람 보다 느리면 기다림
                        # 기다렸던 시간
                        if start < end:
                            wait_time = end - start
                            total_wait_time += wait_time
                            # 이전 사람이 끝난 직후로 저장
                            wating_queue[i].append([end, end + time])
                        else:
                            wating_queue[i].append([start, start + time])
        min_wait_time = min(min_wait_time, total_wait_time)
    return min_wait_time

• 정확성

3. 해설

재도전 필요