BOJ 20006 - 랭킹전 대기열

문제

백준 온라인 저지 - 20006번

풀이 과정

랭킹 대기열 로직을 구현하는 문제입니다.
먼저 방의 정보를 관리하기 위해 해시 테이블 을 사용했습니다.
이때 해당 방에 방장의 식별자와 참여중인 다른 사람들을 배열 형태로 관리합니다.
여기서 assign_player 함수에서는 플레이어를 기존에 존재하는 방에 할당 가능한지 판단하고
만약 새로운 방이 필요할 경우 방장으로 할당한 뒤 새로운 방을 생성하도록 합니다.

코드

import sys


def assign_player(player_id, player_level, rooms):
    for _, meta in rooms.items():
        master_level = players[meta['master']]
        if master_level - 10 <= player_level <= master_level + 10 and len(meta['participant']) < M - 1:
            meta['participant'].append(player_id)
            return True

    return False


def solution():
    rooms = {}

    for player_id, player_level in players.items():
        if not assign_player(player_id, int(player_level), rooms):
            rooms[len(rooms)] = {'master': player_id, 'participant': []}

    for room_id in rooms:
        master = rooms[room_id]['master']
        participants = rooms[room_id]['participant']

        if len(participants) == M - 1:
            print('Started!')
        else:
            print('Waiting!')

        for player in sorted(participants + [master]):
            print(f"{players[player]} {player}")


if __name__ == '__main__':
    P, M = list(map(int, sys.stdin.readline().split()))
    players = {}
    for _ in range(P):
        level, id = list(sys.stdin.readline().split())
        players[id] = int(level)

    solution()