結果

問題 No.2075 GCD Subsequence
ユーザー とりゐとりゐ
提出日時 2022-09-16 22:02:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,649 ms / 4,000 ms
コード長 1,280 bytes
コンパイル時間 332 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 118,516 KB
最終ジャッジ日時 2024-12-21 20:17:21
合計ジャッジ時間 21,125 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

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]
      d.append(k)
      while n%k==0:
        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

m=10**6+10
soe=SOE(m)
n=int(input())
a=list(map(int,input().split()))
dp=[0]*m

mod=998244353
ans=0

for ai in a:
  if ai==1:
    ans+=1
    continue
  f=soe.fact(ai)
  s=len(f)

  res=1
  for bit in range(1,1<<s):
    tmp=1
    for i in range(s):
      if (bit>>i)&1:
        tmp*=f[i]
    res+=dp[tmp]
  
  res%=mod
  for bit in range(1,1<<s):
    tmp=1
    sgn=-1
    for i in range(s):
      if (bit>>i)&1:
        tmp*=f[i]
        sgn=-sgn
    dp[tmp]+=sgn*res
    dp[tmp]%=mod

ans+=sum(dp)
print(ans%mod)
0