結果

問題 No.714 回転寿司屋のシミュレート
ユーザー cherrypi59
提出日時 2018-07-13 23:07:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 894 bytes
コンパイル時間 437 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-10-09 05:32:30
合計ジャッジ時間 2,784 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6 WA * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq


def main():
    n = int(input())
    sd = {}
    seat = [0 for i in range(20)]

    for i in range(n):
        data = input().split()
        if data[0] == "0":
            seat[int(data[1])-1] = 1
            for j in range(int(data[2])):
                if data[3+j] not in sd:
                    sd[data[3+j]] = []
                heapq.heappush(sd[data[3+j]], int(data[1])-1)

        elif data[0] == "1":
            if data[1] in sd:
                while len(sd[data[1]]) != 0 and seat[sd[data[1]][0]] == 0:
                    heapq.heappop(sd[data[1]])
                if len(sd[data[1]]) == 0:
                    print(-1)
                else:
                    idx = heapq.heappop(sd[data[1]])
                    print(idx+1)
            else:
                print(-1)

        else:
            seat[int(data[1])-1] = 0


if __name__ == '__main__':
    main()
0