結果

問題 No.1242 高橋君とすごろく
ユーザー kimiyukikimiyuki
提出日時 2020-10-02 21:58:51
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 986 bytes
コンパイル時間 91 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,520 KB
最終ジャッジ日時 2024-07-20 02:44:00
合計ジャッジ時間 1,958 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
from typing import *

def solve(n: int, k: int, a: List[int]) -> bool:
    b = set(a)
    for a_i in reversed(a):
        for i in reversed(range(max(1, a_i - 100), a_i + 1)):
            if ((i + 1) in b and (i + 6) in b) or ((i + 2) in b and (i + 5) in b) or ((i + 3) in b and (i + 4) in b):
                b.add(i)
            if (i + 1) in b and (i + 6) in b and (i + 2) in b and (i + 5) in b and (i + 3) in b and (i + 4) in b:
                return False
    return 1 not in b


YES = 'Yes'
NO = 'No'


# generated by online-judge-template-generator v4.7.1 (https://github.com/online-judge-tools/template-generator)
def main():
    import sys
    tokens = iter(sys.stdin.read().split())
    N = int(next(tokens))
    K = int(next(tokens))
    A = [None for _ in range(K)]
    for i in range(K):
        A[i] = int(next(tokens))
    assert next(tokens, None) is None
    a = solve(N, K, A)
    print(a and YES or NO)


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