結果

問題 No.689 E869120 and Constructing Array 3
ユーザー maspymaspy
提出日時 2020-03-04 16:30:31
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 480 ms / 1,000 ms
コード長 1,135 bytes
コンパイル時間 71 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 43,860 KB
最終ジャッジ日時 2024-10-14 00:18:51
合計ジャッジ時間 9,129 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 464 ms
43,492 KB
testcase_01 AC 480 ms
43,740 KB
testcase_02 AC 468 ms
43,588 KB
testcase_03 AC 468 ms
43,496 KB
testcase_04 AC 463 ms
43,596 KB
testcase_05 AC 465 ms
43,860 KB
testcase_06 AC 457 ms
43,584 KB
testcase_07 AC 467 ms
43,452 KB
testcase_08 AC 467 ms
43,500 KB
testcase_09 AC 469 ms
43,844 KB
testcase_10 AC 474 ms
43,844 KB
testcase_11 AC 474 ms
43,736 KB
testcase_12 AC 466 ms
43,500 KB
testcase_13 AC 467 ms
43,860 KB
testcase_14 AC 465 ms
43,456 KB
testcase_15 AC 467 ms
43,736 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
# %%
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines


# %%
import numpy as np


# %%
def prime_table(N):
    """Create prime table of interval [0, N)

    Parameters
    ----------
    N : int
        Upper bound. N must be >= 2.

    Returns
    -------
    tuple of two arrays.
    is_prime: is_prime[i] = True iff i is prime.
    primes: primes[i] = (i+1)-th prime
    """
    is_prime = np.zeros(N, np.bool_)
    is_prime[2] = 1
    is_prime[3::2] = 1
    for p in range(3, N, 2):
        if p * p > N:
            break
        if is_prime[p]:
            is_prime[p * p::p + p] = 0
    primes = np.where(is_prime)[0]
    return is_prime, primes


# %%
K = int(readline())

if K == 0:
    print(1)
    print(1)
    exit()

# %%
is_prime, primes = prime_table(10 ** 4)

# %%
n = 0
while (n + 1) * (n + 2) // 2 <= K and n < 100:
    n += 1

C = [1] * n + [2]
rest = K - n * (n + 1) // 2
k = rest // n
C += [4] * k
rest -= n * k

# %%
primes = primes[~(is_prime[primes + 2])]
C += (primes[10:10 + rest] - 2).tolist()
print(len(C))
print(*C)
0