結果

問題 No.1242 高橋君とすごろく
コンテスト
ユーザー FromBooska
提出日時 2023-09-05 12:24:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,378 bytes
コンパイル時間 192 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 81,280 KB
最終ジャッジ日時 2024-06-23 08:02:20
合計ジャッジ時間 4,332 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 6 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

# すごろくだしゴールから戻っていくか
# そのマスがOKかどうかの判定、1or6, 2or5, 3or4先のすべてが止まれればOKマス
# 戻っていってスタートの1がOKならYes
# コンセプトはそれでいいがN10**18は巨大すぎる
# ただしK<100なのでここは相当非効率でも間に合うと思ったがマルチセットでもTLE
# heapでやる、存在確認用のセットも使う

N, K = map(int, input().split())
A = list(map(int, input().split()))
A_set = set(A)

from heapq import *
H = [-a for a in A]
heapify(H)
ans = 'Yes'
while H:
    largest = -heappop(H)
    if largest == 1:
        ans = 'No'
    
    A_set.discard(largest)
    if largest-6 >= 1 and largest-5 in A_set and largest-6 not in A_set:
        heappush(H, -(largest-6))
        A_set.add(largest-6)
    if largest-5 >= 1 and largest-3 in A_set and largest-5 not in A_set:
        heappush(H, -(largest-5))
        A_set.add(largest-5)
        # このケースだけは新たにNGが生まれうる
        if largest-6 >= 1 and largest-5 in A_set and largest-6 not in A_set:
            heappush(H, -(largest-6))
            A_set.add(largest-6)
    if largest-4 >= 1 and largest-1 in A_set and largest-4 not in A_set:
        heappush(H, -(largest-4))
        A_set.add(largest-4)
    #print('largest', largest, 'H', H, 'A_set', A_set)
print(ans)
0