結果

問題 No.1791 Repeat Multiplication
ユーザー とりゐ
提出日時 2021-12-20 22:33:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,444 ms / 3,000 ms
コード長 1,048 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 82,036 KB
実行使用メモリ 118,144 KB
最終ジャッジ日時 2024-09-15 15:21:59
合計ジャッジ時間 31,098 ms
ジャッジサーバーID
(参考情報)
judge6 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

# 貰う DP

# Sieve of Eratosthenes
class SOE:
  def __init__(self,m):
    self.sieve=[-1]*(m+1)
    self.prime=[]
    for i in range(2,m+1):
      if self.sieve[i]==-1:
        self.prime.append(i)
        self.sieve[i]=i
        j=2*i
        while j<=m:
          self.sieve[j]=i
          j+=i
  
  def primes(self):
    # get primes
    return self.prime
  
  def fact(self,n):
    # prime factorization
    d={}
    while n!=1:
      k=self.sieve[n]
      if k not in d:
        d[k]=0
      d[k]+=1
      n//=k
    return d
  
  def div(self,n):
    # get divisors
    c=[1]
    while n!=1:
      p=self.sieve[n]
      cnt=1
      n//=p
      while self.sieve[n]==p:
        cnt+=1
        n//=p
      s=c.copy()
      for i in s:
        for j in range(1,cnt+1):
          c.append(i*(p**j))
    return c

n,q=map(int,input().split())
soe=SOE(n)
dp=[0]*(n+1)
dp[1]=1
sm=[0]*(n+1)
for i in range(1,n+1):
  for j in soe.div(i):
    if i!=j:
      dp[i]+=dp[j]
  sm[i]=dp[i]+sm[i-1]
for _ in range(q):
  x=int(input())
  print(dp[x]*sm[n//x])
0