結果
| 問題 | 
                            No.1242 高橋君とすごろく
                             | 
                    
| コンテスト | |
| ユーザー | 
                             AEn
                         | 
                    
| 提出日時 | 2022-12-03 16:18:37 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 41 ms / 2,000 ms | 
| コード長 | 660 bytes | 
| コンパイル時間 | 189 ms | 
| コンパイル使用メモリ | 82,448 KB | 
| 実行使用メモリ | 54,652 KB | 
| 最終ジャッジ日時 | 2024-10-10 18:32:43 | 
| 合計ジャッジ時間 | 2,318 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 24 | 
ソースコード
from heapq import heappop, heappush
N, K = map(int, input().split())
A = list(map(int, input().split()))
h = []
over = set(A)
for i in range(K):
    heappush(h,-A[i])
cnt,bf = 0, 1<<60
while h:
    v = -heappop(h)
    if v+1==bf:cnt+=1
    else:cnt=1
    bf = v
    if cnt>=4:exit(print('No'))
    # 1,6
    if v+5 in over and v-1 not in over and v-1>0:
        over.add(v-1)
        heappush(h,-(v-1))
    # 2,5
    if v+3 in over and v-2 not in over and v-2>0:
        over.add(v-2)
        heappush(h,-(v-2))
    # 3,4
    if v+1 in over and v-3 not in over and v-3>0:
        over.add(v-3)
        heappush(h,-(v-3))
print('No' if 1 in over else 'Yes')
            
            
            
        
            
AEn