結果

問題 No.1514 Squared Matching
コンテスト
ユーザー 👑 Kazun
提出日時 2021-05-22 04:30:23
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,092 bytes
コンパイル時間 284 ms
コンパイル使用メモリ 82,076 KB
実行使用メモリ 843,200 KB
最終ジャッジ日時 2024-10-10 10:38:33
合計ジャッジ時間 34,951 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 9 MLE * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#エラトステネスの篩
def Sieve_of_Eratosthenes(N,mode=False):
    """Nまでのエラトステネスの篩を実行

    N:自然数
    mode:False->素数のリスト,True->素数かどうかのリスト
    (False->[2,3,5,...],True->[0,0,1,1,0,1,...])
    """

    if N==0:
        return [0]

    T=[1]*(N+1)
    T[0]=T[1]=0

    for x in range(4,N+1,2): T[x]=0
    for x in range(9,N+1,3): T[x]=0

    a=5
    Flag=0
    while a*a<=N:
        if T[a]:
            b=a*a
            c=2*a
            while b<=N:
                T[b]=0
                b+=c
        a+=2+2*Flag
        Flag^=1

    if mode:
        return T
    else:
        return [k for k in range(N+1) if T[k]]
#==================================================
N=int(input())
M=0
while M*M<=N:
    M+=1

Y=[0]*(N+1)
A=list(range(N+1))

for p in Sieve_of_Eratosthenes(M):
    if p*p>N: break

    q=s=p*p
    Y[q]=1
    while q<=N:
        r=q
        while r<=N:
            A[r]//=s
            r+=q
        q*=p*p

for i in range(1,N+1): Y[i]+=Y[i-1]

K=0
for i in range(1,N+1):
    K+=Y[N//A[i]]

print(K)
0