結果

問題 No.714 回転寿司屋のシミュレート
ユーザー mitsuo
提出日時 2018-08-20 14:41:54
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 1,044 bytes
コンパイル時間 81 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-11-25 02:03:10
合計ジャッジ時間 2,224 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

def solve(N, lines):
    hopelist = {}
    result = []
    for seatn in range(1, 21):
        hopelist.setdefault(seatn, [])

    for l in lines:
        if l.startswith("0"):
            d = l.split(" ")
            seat = int(d[1])
            _ = int(d[2])
            hopelist[seat] = d[3:]
        elif l.startswith("1"):
            neta = l.split(" ")[1]
            found = False

            for seatn in range(1, 21):
                hopes = hopelist[seatn]
                for i, h in enumerate(hopes):
                    if h == neta:
                        hopes.pop(i)
                        result.append(seatn)
                        found = True
                        break
                if found:
                    break

            if not found:
                result.append(-1)
        elif l.startswith("2"):
            seat = int(l.split(" ")[1])
            hopelist[seat] = []

    return result

if __name__ == "__main__":
    N = int(input())
    [print(s) for s in solve(N, [input() for _ in range(N)])]
0