結果

問題 No.7 プライムナンバーゲーム
コンテスト
ユーザー ryusuke
提出日時 2023-07-01 00:52:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 132 ms / 5,000 ms
コード長 923 bytes
コンパイル時間 927 ms
コンパイル使用メモリ 82,136 KB
実行使用メモリ 63,796 KB
最終ジャッジ日時 2024-07-07 12:02:23
合計ジャッジ時間 2,455 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

# verification-helper: PROBLEM https://yukicoder.me/problems/no/7

def primes(n: int) -> list:
    "素数の列挙を行う"

    is_prime = [True] * (n + 1)
    is_prime[0] = False
    is_prime[1] = False

    for i in range(2, int(n ** 0.5) + 1):
        if not is_prime[i]:
            continue
        for j in range(i * 2, n + 1, i):
            is_prime[j] = False

    return [i for i, j in enumerate(is_prime) if j]


def main() -> None:

    n = int(input())
    primes_list = primes(n)

    # dp[i] := 残りが i の時にあなたが勝つか
    dp = [False] * (n + 1)
    dp[0] = True
    dp[1] = True
    for i in range(2, n + 1):
        check = False
        for j in primes_list:
            if 0 <= i - j <= n:
                if dp[i - j] == False:
                    check = True
        
        dp[i] = check
    
    print("Win") if dp[n] else print("Lose")


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