結果

問題 No.458 異なる素数の和
ユーザー Yuu EguciYuu Eguci
提出日時 2020-12-08 15:00:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,099 ms / 2,000 ms
コード長 2,875 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 81,640 KB
実行使用メモリ 416,768 KB
最終ジャッジ日時 2024-09-17 14:40:56
合計ジャッジ時間 8,052 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 47 ms
61,440 KB
testcase_01 AC 356 ms
176,520 KB
testcase_02 AC 447 ms
204,152 KB
testcase_03 AC 111 ms
89,984 KB
testcase_04 AC 136 ms
90,112 KB
testcase_05 AC 931 ms
357,760 KB
testcase_06 AC 427 ms
200,280 KB
testcase_07 AC 47 ms
62,976 KB
testcase_08 AC 912 ms
360,064 KB
testcase_09 AC 66 ms
71,424 KB
testcase_10 AC 34 ms
52,224 KB
testcase_11 AC 1,099 ms
416,768 KB
testcase_12 AC 33 ms
52,096 KB
testcase_13 AC 33 ms
52,352 KB
testcase_14 AC 32 ms
51,968 KB
testcase_15 AC 33 ms
51,968 KB
testcase_16 AC 91 ms
88,688 KB
testcase_17 AC 40 ms
58,240 KB
testcase_18 AC 41 ms
58,876 KB
testcase_19 AC 33 ms
52,224 KB
testcase_20 AC 46 ms
59,520 KB
testcase_21 AC 34 ms
52,224 KB
testcase_22 AC 36 ms
51,840 KB
testcase_23 AC 42 ms
59,904 KB
testcase_24 AC 43 ms
59,520 KB
testcase_25 AC 36 ms
52,864 KB
testcase_26 AC 39 ms
57,984 KB
testcase_27 AC 396 ms
181,504 KB
testcase_28 AC 1,067 ms
409,216 KB
testcase_29 AC 57 ms
68,172 KB
testcase_30 AC 268 ms
135,680 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import sqrt


def prime(N: int):
    """問題を解きます。"""

    # 素数一覧。
    primes = get_primes_until_n(N)
    # pprint(primes)

    # 表のいちばん左の列。行の数は、冒頭の例でいえば 18+1。重量 0 のときの行があるからひとつ多い。
    # -1 は、「その重さは作れない」ことを意味します。
    # NOTE: メモリ効率のためリストオブジェクトは繰り返し使います。
    _ = [0] + [-1] * N
    dp = [_ * 1 for i in range(len(primes) + 1)]

    # 列を埋めていきます。
    for i in range(len(primes)):

        # i の周回では primes[i] まで選べるときの計算をします。
        # NOTE: 表でいえば i+1 列目ね。
        value = 1
        weight = primes[i]

        # 行を埋めていきます。
        for w in range(N + 1):

            # 現在のマスに素数を足した場合を検証します。
            # が、今は「作れない」マスならスキップ。
            # 足して N を超えるなら問題外なのでこれもスキップ。
            if dp[i][w] != -1 and w + weight <= N:
                dp[i + 1][w + weight] = dp[i][w] + value

            # あとひとつ右のマスも更新しておきます。
            dp[i + 1][w] = max(dp[i][w], dp[i + 1][w])

    # dp テーブル作り終わりました。
    # この表には、素数が i 個選べるとき w をぴったり作れる素数の最大個数が入っています。
    # pprint(dp)
    # 一番右下の数が、「素数リストから好きなだけ選んで」「N をぴったり作れる」素数の最大個数です。
    return dp[-1][-1]


def get_primes_until_n(N: int):
    """N までの素数一覧を返します。"""

    # 2〜N の dictionary です。
    # この先の処理で、素数でないものは False にしていきます。最終的に値が True のまま残った key が素数です。
    # NOTE: list にして、 index を key 扱いしたほうがイカすんだけど dictionary のほうがわかりやすいかと思って。
    dic = {i: True for i in range(2, N + 1)}

    # N の平方根までチェックすれば、全部の数の素数判定は終わります。
    for i in range(2, int(sqrt(N)) + 1):

        # すでに False(素数ではない)判定になっているものは計算不要です。
        if dic[i] is False:
            continue

        # 2 から始まるので、その先の倍数を全部 False(素数ではない)にしていけば最後には素数だけが True で残ります。
        j = i * 2
        while j <= N:
            dic[j] = False
            j += i

    return [i for i in dic.keys() if dic[i]]


# print(prime(18) == 3)
# print(prime(4) == -1)
# print(prime(3) == 1)
# print(prime(3344) == 41)
print(prime(int(input())))
0