結果

問題 No.1242 高橋君とすごろく
ユーザー FromBooska
提出日時 2023-03-20 19:23:51
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,990 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 82,164 KB
実行使用メモリ 381,744 KB
最終ジャッジ日時 2024-09-18 14:09:58
合計ジャッジ時間 4,145 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 6 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

# あるポジション0から見て(1, 6)両方がアウトならアウト、同様に(2, 5), (3, 4)
# ということはダメなマスがあったら、そこから右に+1, +3, +5がダメなら、左に新しいダメマスができる
# そうやってダメマスを左に作っていき、スタートの1がダメマスならNo, else Yes
# ダメマスは最初のK個からどんどん増えていって、最大値のものから調べる必要がある
# マイナスにしてheapするか
# TLEしたので、heapに加えるかどうかのsetも加えてみる
# それでもTLE、main()を使うか、それともDame数が3連続となったらそれで終わりとするか
from heapq import *

def main():
    
    N, K = map(int, input().split())
    A = list(map(int, input().split()))
    Dame = [-a for a in A]
    Dame_set = set(Dame)
    heapify(Dame)
    visited = set()

    while Dame:
        largest = heappop(Dame)
        Dame_set.discard(largest)

        if (largest-1) in visited and (largest+3) < 0 and (largest+3) not in Dame_set:
            heappush(Dame, largest+3)
            Dame_set.add(largest+3)
            if largest+4 in Dame_set and largest+5 in Dame_set:
                print('No')
                exit()
        if (largest-3) in visited and (largest+2) < 0 and (largest+2) not in Dame_set:
            heappush(Dame, largest+2)
            Dame_set.add(largest+2)
            if largest+3 in Dame_set and largest+4 in Dame_set:
                print('No')
                exit()
        if (largest-5) in visited and (largest+1) < 0 and (largest+1) not in Dame_set:
            heappush(Dame, largest+1)
            Dame_set.add(largest+1)
            if largest+2 in Dame_set and largest+3 in Dame_set:
                print('No')
                exit()
        visited.add(largest)
        #print('Dame', Dame, 'Dame_set', Dame_set, 'visited', visited)
    if (-1) in visited:
        print('No')
    else:
        print('Yes')

main()
0